Ubuntu安装NVIDA、CUDA、cuDNN

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

sudo apt-get install gcc

sudo apt-get install g++

sudo apt-get install make

sudo apt-get update

2.安装NVIDA

1）去NVIDA官网查找并下载相应的显卡驱动.run文件

卸载原有驱动的方法（若未安装过驱动可跳过）

#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

lsmod | grep nouveau

观察是否有输出
若有输出

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

在最后一行添加

blacklist nouveau

之后

sudo update-initramfs -u

重启

输入

lsmod | grep nouveau

若没有输出，则证明成功

3)安装驱动

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

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

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

先赋予权限

sudo chmod a+x *.run

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

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

参数说明

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

安装完成后reboot重启

安装完成后，输入

nvidia-smi

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

4）一些注意事项

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

3.安装CUDA

进入CUDA官网下载相应版本的.run文件

根据官网提示安装CUDA的.run文件

sudo sh cuda_*.run

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


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

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

10.1+版本在安装界面发生了很大的改变，在选择安装界面，应修改为：

如果想要修改cuda的安装目录，则在Options内进行配置，记住，在修改了cuda的安装目录后，后面的环境配置的目录也要相应的修改。

若出现missing recommended libraries错误

安装依赖

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

之后再次安装即可

配置环境变量

打开.bashrc文件

sudo vim ~/.bashrc

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

在文件结尾加上

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文件生效

source ~/.bashrc

验证

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

卸载

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

sudo ./uninstall_cuda_*.pl

4.安装cuDNN

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

选择cuDNN Library for Linux下载

参考cuDNN Installation Guide进行安装

解压下载的.tgz文件

tar -xzvf cudnn-*.tgz

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

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版本找到卸载脚本：

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

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

sudo rm -rf /usr/local/cuda-8.0/