记录windows下安装两个版本tensorflow的过程

记录windows下安装两个版本tensorflow的过程

去年tensorflow出了2.0版本,奈何新版本用不习惯,所以干脆把1.x和2.0版本都装了。

0 准备

本机环境: windows10 + GTX1680
安装方法: anaconda3创建虚拟环境,然后用pip安装

1 anaconda3安装

下载python3.7版本

https://www.anaconda.com/distribution/#download-section

注意:
A.勾选“Install for Just Me (recommended) ”
B.不要勾选“Add Anaconda to my PATH environment variable.”
C.如果不打算使用多个版本的Anaconda或者多个版本的Python,便勾选“Register Anaconda as my default Python 3.7”。

验证:
打开Anaconda Prompt,输入

conda --version

若能显示conda版本号,则安装成功

2 安装tensorflow2.0.0a(GPU版本)

打开Anaconda Prompt

创建虚拟环境
tf2_win是环境名,可替换

conda create -n tf2_win python=3.7

激活环境
左侧出现 (tf2_win) 即表示激活成功

conda activate tf2_win

pip安装
a是内测版本,b是公测版本
如果不打算装gpu版本,把-gpu去掉即可

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==2.0.0a0  

验证是否安装成功
出现:tensorflow-gpu 2.0.0a0 pypi_0 pypi
表示安装成功

conda list 

安装cuda10.0和cudann7.6.0
如果安装CPU版本可以跳过

conda install cudnn=7.6.0
conda install cudatoolkit=10.0.130

验证是否安装成功
出现: cudnn 7.6.0 cuda10.0_0即表示安装成功

 conda list

3 安装tensorflow1.14.0(CPU版本)

和安装2.0版本类似
创建虚拟环境,tf1_win是环境名

conda create -n tf1_win

激活环境

conda activate tf1_win

安装1.14.0 cpu 版本

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow==1.14.0

验证安装成功
出现:
base F:\Anaconda3_win
tf1_win * F:\Anaconda3_win\envs\tf1_win
tf2_win F:\Anaconda3_win\envs\tf2_win

conda env list

4 之后遇到的问题

打开Jupyter notebook运行代码时报错

AttributeError: ‘Tensor’ object has no attribute ‘numpy’

解决方法:重装!
进入虚拟环境后

pip uninstall numpy
pip install -U numpy

5 参考

这是我第一次发文,参考了不少大神的文章。

https://www.jianshu.com/p/62f155eb6ac5
https://www.cnblogs.com/pengweiblog/p/11289674.html
https://zhuanlan.zhihu.com/p/71581533

以上方法亲测可用,欢迎交流。

发布了1 篇原创文章 · 获赞 1 · 访问量 61

猜你喜欢

转载自blog.csdn.net/weixin_43470383/article/details/104064540