Linux下安装cuda和对应版本的cudnn
1.首先在安装cuda与cudnn之前,系统需要成功安装Nvidia驱动,安装教程请参照以下教程:
2.验证系统内部是否已经安装了cuda
打开命令行,输入以下命令:
nvcc -V
若出现以下输出,则系统内部没有安装cuda。
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安装文件
输入上述命令后,稍有卡顿,随后出现以下界面:
输入accept进行后续操作,随即出现以下界面:
因为我们在安装cuda之前已经安装了Nvidia的驱动,因此这里的第一项我们必须取消勾选,选择不安装驱动,随后选择Install进行后续操作。
后续出现的一系列选项,我们都可以选择yes,最终出现以下界面:
随后,我们开始进行环境变量的配置:
打开命令行输入以下命令进行~/.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安装成功:
6.多个cuda版本切换的方法
此种安装方式,是将/usr/local的cuda文件链接到了对应cuda版本的根目录。输入以下命令可以查看当前cuda文件的链接情况:
cd /usr/local/
stat cuda
随后输出以下信息:
在/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版本,已经更改成功,更换其他版本同样操作即可:
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全部安装成功。
更多推荐
所有评论(0)