刚买的新电脑,然后打算配个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,安装基本流程就是一个个下一步了,要注意的是配置环境,只需要把下图的两个勾勾上就可以
二:第二大步就是安装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
(2)CuDNN下载地址:https://developer.nvidia.com/rdp/cudnn-download下载CuDNN需要注册账号
(3)安装cuda和cuDNN的步骤可以参考https://blog.csdn.net/weixin_39290638/article/details/80045236
我这一步选的是精简安装,(安装位置不要换)我的安装步骤是如果遇到下图这种情况需要去官网更新一下驱动
官网网址:http://www.geforce.cn/drivers
你可以下载GeForce Experience让它自己帮你安装(我是最开始没有卸载更新驱动最后安装失败)
安装成功的界面大概是下图这样子的
(4).验证CUDA安装成功:
打开命令行,也就是cmd然后输入“nvcc -V”,如果安装正确的话你应该看到这样的输出:
(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
安装时选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)
出现这个则表明安装成功
建议参考:
我主要是按照以下几个博客操作的:
http://www.cnblogs.com/gaofighting/p/9917456.html#_labelTop
cuda相关操作按照:https://blog.csdn.net/weixin_39290638/article/details/80045236
写这个博客主要记录一下自己的问题,以便自己后面遇到错误重新安装,图片没有及时截取,博客上的图片多为借鉴其他博客,感谢