全套版本:win10+anaconda3+cuda9.0+cudnn7.3.1+tensorflow-gpu1.10.0
实验室电脑配置不高,下载了好几天,出现了很多问题,几度崩溃,最后还是成功了……好像。把流程捋一遍。
注1:不管之前安没安装过tensorflow cpu版,都从第一步开始走。
①安装cuda9.0
我的显卡是GeForce GTX 1050,我选择安装cuda9.0,这个资源网上有很多,如果有需要可以留言,我做一个百度网盘。
安装目录的选择:默认C盘,我尝试了一次安装在其他盘,但是安装失败了,默认C盘路径就不会出错。
精简版/自定义版的选择:选择自定义版,网上大部分教程是全选,我只略去了GeForce Experience部分,其余全选。
②安装cudnn7.3.1
网上资源包二者都是配套的
cudnn-9.0-windows10-x64-v7.3.1.20 这是文件名称。
将里面的文件复制粘贴到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0文件夹里。
③anaconda新建环境
在anaconda prompt中输入命令
conda create -n env_name python=3.6
里面env_name是自己定义的环境名称。
④激活环境
继上一步输入
activate env_name
表示我们进入了这个环境,安装的过程都是在特定的环境中进行的。
注2:这四步可以1234,也可以3124.
⑤下载tensorflow-gpu
我下载过很多次,换了很多不同的版本,大概是我查找资料的能力有限,很多配套版本都讲的不清不楚的,有的版本运行程序之后会提示有问题,最后我选择1.10.0这个版本。
继续在anaconda prompt中输入
pip install tensorflow-gpu==1.10.0
注3:如果提示pip需要更新,就按照提示的命令进行更新。
注4:这个命令有时候会因为网速的问题失败,本来下载的时候速度就会很慢,我中间失败过一次,大概是因为电脑一段时间没动之后自动黑屏要睡眠了,我马上动鼠标亮屏之后就提示失败了,我就再输入一遍命令,注意不要让电脑黑屏,网速最好好一点。
⑥安装完成,测试
在该环境中继续输入命令:
python
会进入到python中,这时输入代码:
import tensorflow as tf
hello=tf.constant('Hello,Tensorflow!')
sess=tf.Session()
print(sess.run(hello))
输出结果就成功了。如果还出现了其他的提示,请仔细阅读,有的是说明GPU情况,也可能是别的问题,但是GPU部分已经安装完成了。
⑦附
退出python的代码:
exit()
退出环境的命令:
deactivate env_name
在查资料的过程中,我还看到另一种安装命令比较有效,如下:
conda create -n env_name tensorflow-gpu
直接新建了环境也安装了tensorflow-gpu,但是版本应该是默认最新的,会不会出现匹配问题仍未知。
cuda8.0/9.0的问题:
我见有人说二者可以共存,但要先下载低版本再下载高版本,也有人说下载新的cuda时,之前的要删干净。我在两个版本共存时还是出现了一些问题的,但是原因未知,只存在9.0时成功。