1. CUDA软件
Windows
打开NVIDIA CUDA网站,选择需要下载的版本,依次选择Windows平台,x86_64架构,10系统,exe(local)本地安装包,再选择Download即可下载CUDA安装软件。
下载完成后,打开软件,选择自定义,按下图进行设置
点击下一步,直至安装结束
打开cmd命令行,输入nvcc -V,能够打印当前CUDA版本信息,说明安装成功
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Sun_Jul_28_19:12:52_Pacific_Daylight_Time_2019
Cuda compilation tools, release 10.1, V10.1.243
2. cuDNN神经网络加速库
CUDA并不是针对神经网络设计的GPU加速库,如果希望针对神经网络应用加速,需要额外安装cuDNN。打开cuDNN网站,先登录账号,填写表格后,进入cuDNN下载界面,勾选“I Agree To the Terms of the cuDNN Software License Agreement”,再选择对于CUDA版本的相应版本进行下载。解压下载的文件,将名为“cuda”的文件夹重命名为“cudnn765”,再将它复制到CUDA下:C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.0
完成上述操作后,还需要让系统感知到cuDNN文件的位置,因此需要额外配置Path环境遍历。打开文件浏览器(快捷键Alt+E),右击“我的电脑”,依次选择“属性”-->“高级系统属性”-->“高级”-->“环境变量”,在“系统变量”一栏中选中“Path”环境变量,依次选择“编辑”-->“新键”,输入cuDNN的安装路径:C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.1cudnn765in,并通过“上移”按钮将其置顶
最后依次点击确定,关闭所有对话框