Win7下cuda8.0深度学习开发平台

注意:

由于tensorflow-gpu开发平台对所支持的软件版本比较敏感,因此强烈建议采用与本博文一致的软件版本,具体如下:

Visual Stdio 2015 + WinPython-64bit-3.5.4.0Qt5 + tensorflow1.4.0 + tensorlayer1.7.0  + cuda8.0.44 + cudnn-8.0-windows7-x64-v6.0

1. 软件安装

1.1 安装 VS

由于cuda开发需要VS环境支持,因此安装前需先安装Visual Stdio2015。

1.2 Python支持

安装Python科学计算平台,参考博文:《Python科学计算平台(WinPython)》

1.3 安装cuda和cudnn

必须下载对应版本的 cuda 和 cudnn,否则无法正常运行。

1.3.1 下载CUDA

下载cuda8.0.44版本,访问官方地址

图片

注:若需下载其它版本的cuda,可访问 地址,下载后,双击安装即可,建议不要更改默认目录。

1.3.2 下载CUDNN

下载cudnn-8.0-windows7-x64-v6.0版本,访问官方地址

图片

注:下载cudnn-8.0-windows7-x64-v6.0时,需要在Nvidia官网需要注册,正常注册即可,下载时注意版本号对应。

2. 软件配置

2.1 更换pip源

在配置winPython-64bit-3.5.4.0Qt5前,建议先配置下pip的源,提高下载速度。

在win7的地址栏中输入如下命令:

%appdata%

在跳转到的目录下创建文件夹pip,并在此文件夹下新建pip.ini文件,添加如下内容:

[global]
index-url = https://pypi.doubanio.com/simple

图片

图片

2.2 安装tensorflow

在WinPython-64bit-3.5.4.0Qt5目录下,打开 WinPython Command Prompt.exe,如下:

图片

  1. 卸载默认的tensorflow版本:
pip uninstall tensorflow
  1. 安装tensorflow-gpu版本:
pip install tensorflow-gpu==1.4.0
pip install tensorlayer==1.7.0
  1. 查看tensorflow-gpu是否安装:
pip list

图片

看到 tensorflow-gpu 输出如上内容,则表示安装完成。

2.3 查看版本

安装cuda8.0.44后,打开cmd通过如下命令查看版本信息:

nvcc -V

图片

2.4 配置cuda

将下载后的cudnn-8.0-windows7-x64-v6.0解压,并将文件夹下bin、include、lib三个文件夹复制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0文件夹下,如下:

图片

2.5 编辑器使用

打开 C:\WinPython-64bit-3.5.4.0Qt5 目录下的 Spyder.exe,创建对应的开发文件,并添加逻辑代码后,运行即可,此时会自动调用 tensorflow-gpu 的API进行运算。