安装 tensorflow2.1
前提条件已经安装了 anaconda 3.7 , 由于网上这部分教程超级多,所以就不再这里赘述了。
Step 1 为了不与其他的现有编译器产生冲突,所以新建立一个专门用来运行tensorflow2.1
的环境。
按照上图打开命令窗口,键入conda create -n Tensorflow21 python=3.7
注:这里 Tensorflow21 就是未来你编译器名称,可以是任意名称。
键入conda create -n Tensorflow21 python=3.7
会出现下图界面,直接输入y。
完成后如下:
Step 2 激活环境
键入conda activate Tensorflow21
Step 3 键入conda install cudatoolkit=10.1
键入后会出现下图,直接y,如果这步报错,很可能是因为电脑没有gpu
。如果是没有gpu
那就可以直接跳过。
这步安装相对较慢,且有可能会断开,这时需要重复Step 3 。
注: 如何查看自己是否有gpu
,网上方法很多,这里给出一种供参考。直接进入任务管理器——性能,如果可以看到gpu
。 那就是有。
Step 4 键入conda install cudnn=7.6
如果这步报错,也是电脑不支持。跳过即可,注意这里看一下不成功是不是因为网络问题导致链接断开,而不是真的电脑不支持。
Step 5 键入pip install tensorflow==2.1
Step 6 验证是否安装成功。
键入python 进入python环境 import tensorflow as tf
如果出现绿色框内容证明成功。
再输入tf.__version__
如果得到2.1.0证明成功安装tensorflow2.1.0
。