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 环境变量设置
打开文件资源管理器,在这台电脑上右击选择属性,在系统中选择高级系统设置
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即可