Win7下cuda8.0深度学习开发平台
admin 于 2017年12月03日 发表在 机器学习笔记

注意:

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

Visual Stdio2015 + WinPython-64bit-3.5.4.0Qt5 + tensorflow1.4.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

(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进行运算。

注意:本站所有文章除特别说明外,均为原创,转载请务必以超链接方式并注明作者出处。

标签:机器学习