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即可

posted on 2022-07-28 09:46  海恺  阅读(596)  评论(0编辑  收藏  举报