ubuntu18.04安装CUDA和cuDNN

2021-04-05 20:29:22 阅读：502 来源： 互联网

标签：10.1 usr cuDNN CUDA ubuntu18.04 local 链接 cuda

安装CUDA和cuDNN

升级驱动

首先将显卡驱动升级到最新版本。（如果没有安装驱动，可能这一步会失败，请先到Nvidia的驱动官网下载。）在shell中输入

sudo apt-get update
sudo apt-get upgrade

安装CUDA

下载CUDA，这里CUDA10.1为例子。首先进入Nvidia的CUDA官网，选择CUDA10.1版本。
在这里插入图片描述
也可以复制下载的链接，直接在Google Chrome里面下载。下载好以后进入到文件所在的路径下面，输入如下命令

sudo sh cuda_10.1.243_418.87.00_linux.run --override --silent --toolkit

安装cuDNN

进入到cuDNN官网，选择和CUDA版本相匹配的版本，这里以cuDNN v8.0.5为例，选择cuDNN Library for Linux下载。注意这里需要登录，没有账号的需要注册一个。
在这里插入图片描述
下载完成后会得到一个.tgz的压缩包，解压得到一个名为cuda的目录，进入目录所在路径，执行以下命令

sudo cp cuda/include/* /usr/local/cuda-10.1/include
sudo cp cuda/lib64/* /usr/local/cuda-10.1/lib64
sudo chmod a+r -R /usr/local/cuda-10.1/include/* /usr/local/cuda-10.1/lib64/*

添加环境变量

输入以下命令

vim ~/.bashrc

在末尾添加如下内容

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda

保存后退出，在terminal中输入如下命令，使得添加的内容生效

source ~/.bashrc
sudo ldconfig

在terminal中输入如下命令，查看是否添加成功。

echo $CUDA_HOME

如果成功输出路径，则表明添加成功。

添加链接

一般情况下，安装CUDA的时候，会在cuda-10.1文件夹同级的路径下面添加其链接文件，也就是在/etc/local/下会有名为cuda的链接文件。我们在环境变量中添加的路径也就是链接文件的路径。环境变量设置为链接文件的路径有一些好处。首先链接文件只是一个链接，需要制定指向的内容，其本身不占据什么硬盘的空间。其次，我们只需修改链接文件，即可实现CUDA版本的切换。

在相同的路径下我们新建一个脚本，用于创建cuda链接文件。代码如下

#!/bin/bash
sudo rm -rf /usr/local/cuda
sudo ln -s /usr/local/cuda-10.1 /usr/local/cuda

命名为install_cuda_10_1.sh。以后执行该脚本即可完成cuda链接文件的创建。

如果安装了别的版本的CUDA，可以编写其它的链接文件，此时只需将上面代码中的cuda-10.1改为别的cuda文件夹名称，然后保存为另外的脚本即可。需要切换CUDA版本时只需要执行相应的脚本。

在terminal中输入

nvcc --version

卸载CUDA

进入cuda-10.1文件夹，进入bin目录，找到带有uninstall名称的文件执行即可，本人为cuda-10.1/bin/cuda-uninstaller。进入后勾选卸载的内容，执行即可。

执行完成后，可能保留有cuDNN和其它的残余文件，直接将cuda-10.1目录都删除即可。此时会发现，cuda的链接文件已经被删除了，否则自己手动删除该链接文件。

安装时遇到空间不够问题

安装CUDA时空间不够

如果安装CUDA时显示
terminate called after throwing an instance of ‘boost::filesystem::filesystem_error’ what(): boost::filesystem::copy_file: No such file or directory: “./builds/cuda-toolkit/nvml/example/supportedVgpus.c”, “/usr/local/cuda-10.1/nvml/example/supportedVgpus.c” Aborted (core dumped)
则表明根目录不够大，则无法安装。可以删除一些系统日志文件或者安装在/opt下没用的应用，以扩大空间，最后保证起码有8G的空间。

安装cuDNN时发现空间不够

此时不会报错，只是你的根目录已经没有多余的空间了，此时很可能tab键的补全都用不了了。此时如果贸然重启电脑，会陷入开不了机的情况。

如果没有重启，则于上一种情况一样，将cuda卸载，然后清理一些空间再安装。

如果真的重启了，此时不能正确启动，因为没有空间留给系统需要创建的临时文件。解决方法是在grub页面进入recovery mode，选择root。在安全模式下可以进入/usr/local/，卸载CUDA，就能成功启动啦 ~(≧▽≦)/~

标签：10.1,usr,cuDNN,CUDA,ubuntu18.04,local,链接,cuda
来源： https://blog.csdn.net/weixin_43441566/article/details/115445172

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9