tensorflow-gpu安装(win10+anaconda3+GeForce GTX1050)

全套版本: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时成功。

你可能感兴趣的:(软件问题)