Linux下安装cuda和对应版本的cudnn

1.首先在安装cuda与cudnn之前,系统需要成功安装Nvidia驱动,安装教程请参照以下教程:

Nvidia驱动安装教程

2.验证系统内部是否已经安装了cuda

打开命令行,输入以下命令:

nvcc -V

若出现以下输出,则系统内部没有安装cuda。

Linux下安装cuda和对应版本的cudnn_第1张图片

3.进行cuda安装包与cudnn的下载

cuda的下载链接:cuda下载链接

cudnn的下载链接:cudnn下载链接

在下载这两个文件的时候,需要注意cudnn的版本需要与cuda的版本相匹配。

4.进行cuda的安装

在进行cuda安装之前,我们需要先安装cuda的相关依赖库,防止cuda安装出现错误。命令行输入以下命令,进行相关依赖库的安装:

sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev

随后输入以下命令进行cuda的安装:

sudo sh cuda_10.1.105_418.39_linux.run  //其中cuda_10.1.105_418.39_linux.run是我们下载的cuda安装文件

输入上述命令后,稍有卡顿,随后出现以下界面:

Linux下安装cuda和对应版本的cudnn_第2张图片

输入accept进行后续操作,随即出现以下界面:

Linux下安装cuda和对应版本的cudnn_第3张图片

因为我们在安装cuda之前已经安装了Nvidia的驱动,因此这里的第一项我们必须取消勾选,选择不安装驱动,随后选择Install进行后续操作。

后续出现的一系列选项,我们都可以选择yes,最终出现以下界面:

Linux下安装cuda和对应版本的cudnn_第4张图片

随后,我们开始进行环境变量的配置:

打开命令行输入以下命令进行~/.bashrc文件的修改:

vim ~/.bashrc

如果使用的是zsh,则修改~/.zshrc

在文件的末尾加入下面三行:

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda

随后命令行输入以下命令保存并退出:

:wq!

在添加完环境变量后,需要更新一下环境变量,命令行输入以下命令进行环境变量的更新:

source ~/.bashrc

5.cuda安装结果验证

命令行输入以下命令,验证cuda是否安装成功:

nvcc -V

如果出现以下输出,则cuda安装成功:

Linux下安装cuda和对应版本的cudnn_第5张图片

6.多个cuda版本切换的方法

此种安装方式,是将/usr/local的cuda文件链接到了对应cuda版本的根目录。输入以下命令可以查看当前cuda文件的链接情况:

cd /usr/local/
stat cuda

随后输出以下信息:

Linux下安装cuda和对应版本的cudnn_第6张图片

/usr/local下,我们可以安装多个版本的cuda,这里我列出我电脑内的cuda版本:

命令行输入以下命令,断开cuda文件与cuda-10.1的符号链接:

sudo rm -rf cuda

命令行输入以下命令,建立cuda文件与cuda-9.0的符号链接:

sudo ln -s /usr/local/cuda-9.0 /usr/local/cuda

此时查看cuda版本,已经更改成功,更换其他版本同样操作即可:

Linux下安装cuda和对应版本的cudnn_第7张图片

7.进行cudnn的安装

进入到cudnn下载的安装路径下,命令行输入以下命令进行解压操作:

tar -xzvf cudnn-10.1-linux-x64-v8.0.5.39.tgz //这里cudnn-10.1-linux-x64-v8.0.5.39.tgz是我们下载的cudnn的压缩包

随后在当前路径的命令行终端输入以下三条命令进行cudnn的安装:

sudo cp cuda/include/cudnn.h    /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn*    /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h   /usr/local/cuda/lib64/libcudnn*

至此cuda与cudnn全部安装成功。

你可能感兴趣的:(linux,ubuntu,cuda,cudnn)