下载cuda安装包

这里我用CUDA10.1演示,具体看个人需求选择版本,如果已经安装好显卡驱动,需要根据显卡驱动版本选择对应的CUDA工具包。

CUDA10.1:https://developer.nvidia.com/cuda-10.1-download-archive-base
cuDNN:https://developer.nvidia.com/rdp/cudnn-download

安装cuda

双击打开cuda进行安装。

提示选择cuda的临时解压目录,随意选择即可,安装完成后会自动删除。


这里我选择自定义设置。


我的驱动版本高于CUDA自带的安装驱动,因此这里我没有选择安装显卡驱动,仅选择安装CUDA组件,如果没有安装显卡驱动,要记得勾选上。


这里安装在C盘会占用大概3~4G的储存空间,如果C盘空间紧张可以把安装路径修改到其他磁盘,这里我选择安装在C盘。


等待安装成功即可。


召出终端,输入nvcc -V,显示如下效果表示安装成功。


如果不是如图效果,则需要额外添加环境变量。这里注意需要将我的安装路径修改为自己的安装路径。


选择变量Path,点击编辑,新建如下两个路径。

安装cudnn

解压cudnn压缩包,将文件夹下的所有文件复制到CUDA的安装路径下,即完成了cudnn的部署。