下载 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