Win7下cuda8.0深度学习开发平台
注意:
由于tensorflow-gpu开发平台对所支持的软件版本比较敏感,因此强烈建议采用与本博文一致的软件版本,具体如下:
Visual Stdio2015 + WinPython-64bit-3.5.4.0Qt5 + tensorflow1.4.0 + tensorlayer1.7.0 + cuda8.0.44 + cudnn-8.0-windows7-x64-v6.0
一. 软件下载
1. 安装VS
由于cuda开发需要VS环境支持,因此安装前需先安装Visual Stdio2015。
2. Python支持
安装Python科学计算平台,参考博文:《Python科学计算平台(WinPython)》
3. 下载对应版本的cuda和cudnn
(1)下载CUDA
下载cuda8.0.44版本,访问官方地址。
注:若需下载其它版本的cuda,可访问这个地址;下载后,双击安装即可,建议不要更改默认目录。
(2)下载CUDNN
下载cudnn-8.0-windows7-x64-v6.0版本,访问官方地址。
注:下载cudnn-8.0-windows7-x64-v6.0时,需要在Nvidia官网需要注册,正常注册即可,下载时注意版本号对应。
二. 软件安装和配置
1. 更换pip源
在配置winPython-64bit-3.5.4.0Qt5前,建议先配置下pip的源,提高下载速度。
(1)在win7的地址栏中输入如下命令:
%appdata%
(2)在跳转到的目录下创建文件夹pip,并在此文件夹下新建pip.ini文件,添加如下内容:
[global] index-url = https://pypi.doubanio.com/simple
2. 安装tensorflow
在WinPython-64bit-3.5.4.0Qt5目录下,打开 WinPython Command Prompt.exe,如下:
(1)卸载默认的tensorflow版本:
pip uninstall tensorflow
(2)安装tensorflow-gpu版本:
pip install tensorflow-gpu==1.4.0 pip install tensorlayer==1.7.0
(3)查看tensorflow-gpu是否安装:
pip list
看到 tensorflow-gpu 输出如上内容,则表示安装完成。
3. 查看版本
安装cuda8.0.44后,打开cmd通过如下命令查看版本信息:
nvcc -V
4. 配置cuda
将下载后的cudnn-8.0-windows7-x64-v6.0解压,并将文件夹下bin、include、lib三个文件夹复制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0文件夹下,如下:
5. 编辑器使用
打开 C:\WinPython-64bit-3.5.4.0Qt5 目录下的 Spyder.exe,创建对应的开发文件,并添加逻辑代码后,运行即可,此时会自动调用 tensorflow-gpu 的API进行运算。