下载 cuda和 cudnn
-
了解自己机器的
nvidia显卡驱动支持什么版本的cuda最新的信息参照官网的文档, 下图是我在写这篇博客的最新配置,第一列是cuda 版本,第二列是linux驱动版本, 第三列是windows驱动版本。

-
官网下载 你的机器需要的
cuda版本,下图是官网截图的一部分

-
根据自己的操作系统类型,操作系统版本去选择合适的安装包,这里我的是
linux->ubuntu->18.04->runfile

-
长时间等待
cuda的下载, 下载成功后会有一个cuda*.run的文件,这个就是我们需要安装的文件了 -
选择需要的
cudnn版本,同样的,cudnn要和nvidia显卡驱动以及cuda版本要兼容,具体的信息看官网链接
下图是官网截图的最新版本的support matrix,进入官网可以查看全部。

-
下载对应的
cudnn版本,官网链接 需要登录!!!!
安装cuda
- 保证安装的用户有
root权限 bash cuda***.run- 根据提示安装就可以了,这里注意一点,
cuda本身自带显卡驱动,在选择的时候一定要把这一项取消,否则它会覆盖掉原来的显卡驱动 - 安装完成,系统默认安装在了
/usr/local里,你会发现这里面多了cuda和cudaX.X文件夹,后者是真正的安装结果,cuda只是链接它的。 - 配置环境变量:
vim /etc/bash.bashrc
最后一行加入export PATH=${PATH}:/usr/local/cuda/bin
再加入一行export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/local/cuda/lib64
source /etc/bash.bashrc更新环境变量
6.到这里,cuda就完成安装了
安装 cudnn
- 解压刚才下载的
cudnn压缩包
tar -xzvf cudnn-x.x-linux-x64-v8.x.x.x.tgz会生成一个cuda文件夹 - 复制文件
$ sudo cp cuda/include/cudnn*.h /usr/local/cuda/include # root 用户去掉sudo
$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
- 更改权限
$ sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
安装完成
nvcc -V 查看是否成功
版本不一样?
nvidia-smi 显示的cuda 版本和nvcc -V显示的cuda的版本不一样,不用担心,跑程序的时候用的是nvcc -V 对应的cuda

本文档详细介绍了如何下载并安装CUDA和cuDNN。首先,确认显卡驱动兼容的CUDA版本,然后从NVIDIA官网下载合适的CUDA安装包。在安装过程中,务必取消自带驱动选项。接着,下载与CUDA版本匹配的cuDNN,将其头文件和库文件复制到CUDA安装目录。最后,更新环境变量,使CUDA和cuDNN生效。通过nvcc-V和nvidia-smi命令检查安装是否成功。
2503

被折叠的 条评论
为什么被折叠?



