下载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的部署。
评论(0)
您还未登录,请登录后发表或查看评论