win10+ 1060 +anaconda3 4.3.1+ vs2017+cuda9+cudnn v7.3.1+tensorflow-gpu

刚买的新电脑,然后打算配个gpu跑跑数据 ,接下来言归正题,

tesnorflow有gpu和cpu两个版本,gpu版本需要 CUDA 和 cuDNN 的支持。如果你要安装 GPU 版本,请先确认你的显卡支持 CUDA。我安装的是 GPU 版本,采用 pip 安装方式,所以就以 GPU 安装为例,CPU 版本就不用安装 CUDA 和 cuDNN。

1.我先安装的anaconda,(尽量安装在非系统盘吧)anaconda里面继承了python,我安装的是anaconda3 4.3.1版本,他自带的是python3.6,安装基本流程就是一个个下一步了,要注意的是配置环境,只需要把下图的两个勾勾上就可以

win10+ 1060 +anaconda3 4.3.1+ vs2017+cuda9+cudnn v7.3.1+tensorflow-gpu_第1张图片

二:第二大步就是安装Visual Studio,(最好安装在C盘,我装的时候安装位置是可以选择的)我装的是vs2017,之所以要安装vs2017,因为cuda需要Visual Studio的支持

下载地址:https://www.visualstudio.com/zh-hans/

安装过程中注意安装选项:

三:安装cuda和cudnn

(1)在anaconda下安装cuda;

刚开始用的CUDA8.0没成功,因为CUDA8.0是与VS2017不兼容的,就是说如果想用CUDA8.0,必须使用VS2015以下的VS版本
注意1:安装目录别改,默认在c盘C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA
C:\Users\ADMINI~1\AppData\Local\Temp\CUDA
注意2:安装错了可以重装
 

因为现在的CUDA已经更新到了10版本,但是TensorFLow1.7只支持到9.0版。另外,也要确认CUDA版本是否支持自己的显卡。CUDA下载地址:https://developer.nvidia.com/cuda-toolkit-archive

win10+ 1060 +anaconda3 4.3.1+ vs2017+cuda9+cudnn v7.3.1+tensorflow-gpu_第2张图片

 

(2)CuDNN下载地址:https://developer.nvidia.com/rdp/cudnn-download下载CuDNN需要注册账号

win10+ 1060 +anaconda3 4.3.1+ vs2017+cuda9+cudnn v7.3.1+tensorflow-gpu_第3张图片

 

(3)安装cuda和cuDNN的步骤可以参考https://blog.csdn.net/weixin_39290638/article/details/80045236

我这一步选的是精简安装,(安装位置不要换)我的安装步骤是如果遇到下图这种情况需要去官网更新一下驱动

win10+ 1060 +anaconda3 4.3.1+ vs2017+cuda9+cudnn v7.3.1+tensorflow-gpu_第4张图片

官网网址:http://www.geforce.cn/drivers

你可以下载GeForce Experience让它自己帮你安装(我是最开始没有卸载更新驱动最后安装失败)

win10+ 1060 +anaconda3 4.3.1+ vs2017+cuda9+cudnn v7.3.1+tensorflow-gpu_第5张图片

 

安装成功的界面大概是下图这样子的

win10+ 1060 +anaconda3 4.3.1+ vs2017+cuda9+cudnn v7.3.1+tensorflow-gpu_第6张图片

(4).验证CUDA安装成功:

打开命令行,也就是cmd然后输入“nvcc -V”,如果安装正确的话你应该看到这样的输出:

win10+ 1060 +anaconda3 4.3.1+ vs2017+cuda9+cudnn v7.3.1+tensorflow-gpu_第7张图片

(5)安装cuDNN,就是下载后,进行解压将解压后的三个文件夹bin、include、lib文件夹,将这三个文件夹复制到安装CUDA9.0的路径覆盖原有的文件。本机的安装路径为:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0

(6)配置环境变量

打开环境变量,系统变量的Path添加C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin,C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include,C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64,这三个路径。

       cuda和cudnn安装结束。

四:为anaconda配置Tensorflow环境(python环境为3.6)

(1)conda create -n tensorflow_gpu python=3.6

win10+ 1060 +anaconda3 4.3.1+ vs2017+cuda9+cudnn v7.3.1+tensorflow-gpu_第8张图片

安装时选y,在结束后

我们先切换到创建好的环境中:

activate tensorflow_gpu

不用时退回到root环境使用如下代码:

deactivate tensorflow-gpu

(2)在创建好的环境下,安装一些库

conda install anaconda

然后安装GPU版本的tensorflow

conda install tensorflow-gpu

我在安装自己创建的环境下的一些关键包的时候遇到了下面的问题

CondaHTTPError: HTTP 000 CONNECTION FAILED for url 
Elapsed: -

An HTTP error occurred when trying to retrieve this URL.
HTTP errors are often intermittent, and a simple retry will get you on your way.
ConnectTimeout(MaxRetryError("HTTPSConnectionPool(host='repo.continuum.io', port=443): Max retries exceeded with url: /pkgs/main/win-64/repodata.json.bz2 (Caused by ConnectTimeoutError(, 'Connection to repo.continuum.io timed out. (connect timeout=9.15)'))",),)

发现是我的网络发生的更改(使用的无线网,中间断了两次,恢复时就出现了这个问题)导致的,在下载过程中不要更改网络。

五:测试程序

import tensorflow as tf
a = tf.random_normal((100, 100))
b = tf.random_normal((100, 500))
c = tf.matmul(a, b) #矩阵相乘
sess = tf.InteractiveSession()
sess.run(c)

win10+ 1060 +anaconda3 4.3.1+ vs2017+cuda9+cudnn v7.3.1+tensorflow-gpu_第9张图片

出现这个则表明安装成功

 

建议参考:

我主要是按照以下几个博客操作的:

http://www.cnblogs.com/gaofighting/p/9917456.html#_labelTop

cuda相关操作按照:https://blog.csdn.net/weixin_39290638/article/details/80045236

写这个博客主要记录一下自己的问题,以便自己后面遇到错误重新安装,图片没有及时截取,博客上的图片多为借鉴其他博客,感谢

你可能感兴趣的:(win10+ 1060 +anaconda3 4.3.1+ vs2017+cuda9+cudnn v7.3.1+tensorflow-gpu)