win10 GTX1080配置Python CUDA、CUDNN支持

1 环境

win10 64位

VS2017

CUDA9.0  （TensorFlow_gpu1.6以上版本支持CUDA9.0暂时不支持9.1）

2 下载

打开NVIDIA控制面板，如下图所示。选择“系统信息”--“组件”，找到NVCUDA.DLL信息显示即为显卡支持的CUDA最高版本。

通过查看后面产品名称中数字即为最高支持的CUDA版本，然后从下面链接中下载对应的版本

 

https://developer.nvidia.com/cuda-toolkit-archive

3 安装

3.1随便选择解压路径，安装后会自动删除解压文件

等待即可

3.2 安装选项

选择自定义安装

安装过程可能会出现以下错误，所以安装之前可以在自定义中取消勾选CUDA 下面的visual studio integration选项

然后安装即可

4 环境变量设置

打开文件资源管理器，在这台电脑上右击选择属性，在系统中选择高级系统设置

在高级中点击环境变量

点击系统变量下面的新建，添加以下环境变量：

CUDA_BIN_PATH       %CUDA_PATH%\bin

CUDA_LIB_PATH       %CUDA_PATH%\lib\x64

CUDA_PATH           C:\software\CUDA                 （CUDA安装路径）

CUDA_PATH_V9_1      C:\software\CUDA               （CUDA安装路径）

CUDA_SDK_BIN_PATH   %CUDA_SDK_PATH%\bin\win64

CUDA_SDK_LIB_PATH   %CUDA_SDK_PATH%\common\lib\x64

CUDA_SDK_PATH   C:\software\CUDA                 （CUDA安装路径）

NVCUDASAMPLES_ROOT    C:\software\CUDA       （CUDA安装路径）

NVCUDASAMPLES9_1_ROOT    C:\software\CUDA         （CUDA安装路径）

在系统变量中点开PATH，如下

点击新建添加如下环境变量：

（C:\software\CUDA为安装目录根据安装目录自行修改）

 

C:\software\CUDA\bin

C:\software\CUDA\libnvvp

C:\software\visualstudio\VC\Tools\MSVC\14.13.26128\bin\Hostx64\x64

%CUDA_LIB_PATH%

%CUDA_BIN_PATH%

%CUDA_SDK_LIB_PATH%

%CUDA_SDK_BIN_PATH%

C:\software\CUDA\lib\x64

C:\software\CUDA\common\lib\x64

C:\software\CUDA\bin\win64

添加之后即可。可以安装TensorFlow_gpu等进行尝试，建议使用TensorFlow_gpu1.6或之上的版本

5 安装CUDNN

https://developer.nvidia.com/rdp/cudnn-download从上述链接中下载CUDNN

cudnn下载解压之后，将对应的文件分别拷贝到CUDA  Toolkit中即可：

若为默认安装，则应分别拷贝到的文件夹如下：

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

6验证是否安装成功

cmd中命令nvcc -V即可