avatar

Ubuntu安装NVIDA、CUDA、cuDNN

1.先安装好gcc,g++,make

1
sudo apt-get install gcc
1
sudo apt-get install g++
1
sudo apt-get install make
1
sudo apt-get update

2.安装NVIDA

1)去NVIDA官网查找并下载相应的显卡驱动.run文件
卸载原有驱动的方法(若未安装过驱动可跳过)
1
2
3
4
5
6
#for case1: original driver installed by apt-get:
sudo apt-get remove --purge nvidia*

#for case2: original driver installed by runfile:
sudo chmod +x *.run
sudo ./NVIDIA-Linux-x86_64-384.59.run --uninstall
2)关闭nouveau
1
lsmod | grep nouveau

观察是否有输出
若有输出

1
sudo vim /etc/modprobe.d/blacklist.conf

在最后一行添加

1
blacklist nouveau

之后

1
sudo update-initramfs -u

重启

输入

1
lsmod | grep nouveau

若没有输出,则证明成功

3)安装驱动

按下Ctrl+Alt+F2进入命令行界面

然后关掉图形界面,具体方法可查看这篇博客

进入你下载的.run文件的目录

先赋予权限

1
sudo chmod a+x *.run

运行.run文件进行安装(注意参数)

1
sudo ./*.run –no-opengl-files

参数说明

–no-opengl-files 只安装驱动文件,不安装OpenGL文件。这个参数最重要
–no-x-check 安装驱动时不检查X服务
–no-nouveau-check 安装驱动时不检查nouveau
后面两个参数可不加。
安装过程都默认yes

安装完成后reboot重启

安装完成后,输入

1
nvidia-smi

若有输出,则证明安装成功

4)一些注意事项
  • 在安装前,要关闭电脑的secure boot。进入bios,进入BOOT SETUP,进入Security’,将secure boot 设置为disabled。
  • 若在安装过程中出现gcc,make等词汇,说明gcc,make等未安装

3.安装CUDA

进入CUDA官网下载相应版本的.run文件
根据官网提示安装CUDA的.run文件
1
sudo sh cuda_*.run

在安装过程中除了安装驱动选项选择no,其他选择yes或默认


若安装过程出现tmp挂载盘容量不够,则可使用如下命令

1
sudo sh cuda_*.run --tmpdir=/home/ --override

10.1+版本在安装界面发生了很大的改变,在选择安装界面,应修改为:
在这里插入图片描述
如果想要修改cuda的安装目录,则在Options内进行配置,记住,在修改了cuda的安装目录后,后面的环境配置的目录也要相应的修改。

安装依赖

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

之后再次安装即可

配置环境变量

打开.bashrc文件

1
2
3
4
5
sudo vim ~/.bashrc

如果用的zsh , 打开.zshrc文件
sudo vim ~/.zshrc
下面的文件作相应修改

在文件结尾加上

1
2
export PATH="/usr/local/cuda-10.0/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-10.0/lib64:$LD_LIBRARY_PATH"

其中cuda-10.0应改为相应版本

使bashrc文件生效

1
source ~/.bashrc
验证

输入nvcc -V验证能否查看CUDA版本

卸载

/usr/local/cuda/bin目录下运行cuda自带的卸载工具uninstall_cuda_*.pl

1
sudo ./uninstall_cuda_*.pl

4.安装cuDNN

进入cuDNN下载官网下载相应版本的cuDNN

选择cuDNN Library for Linux下载

参考cuDNN Installation Guide进行安装

解压下载的.tgz文件

1
tar -xzvf cudnn-*.tgz

将解压出的文件拷贝到CUDA安装目录

1
2
3
4
5
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*

5.cuda的卸载

卸载CUDA很简单,一条命令就可以了,主要执行的是CUDA自带的卸载脚本,读者要根据自己的cuda版本找到卸载脚本:

1
sudo /usr/local/cuda-8.0/bin/uninstall_cuda_8.0.pl

卸载之后,还有一些残留的文件夹,之前安装的是CUDA 8.0。可以一并删除:

1
sudo rm -rf /usr/local/cuda-8.0/
文章作者: 白丁
文章链接: http://baidinghub.github.io/2020/04/03/Ubuntu%E5%AE%89%E8%A3%85NVIDA%E3%80%81CUDA%E3%80%81cuDNN/
版权声明: 本博客所有文章除特别声明外,均采用 CC BY-NC-SA 4.0 许可协议。转载请注明来自 BaiDing's blog
打赏
  • 微信
    微信
  • 支付寶
    支付寶

评论