本文已参与「新人创作礼」活动,一起开启掘金创作之路。
一、安装显卡驱动
1. 禁用nouveau
安装NVIDIA驱动前先禁用nouveau,否则会出现冲突 使用以下命令打开文件
sudo gedit /etc/modprobe.d/blacklist.conf
在文末添加以下两句话
blacklist nouveau
options nouveau modeset=0
sudo update-initramfs -u
reboot #重启
开机验证nouveau 禁用是否成功
lsmod | grep nouveau
如果什么都没有输出就说明禁用成功
2. 下载驱动文件并安装
查看显卡版本
lspci | grep -i nvidia
查看电脑适合的版本
sudo apt-cache search nvidia*
- 下载驱动文件
驱动官网地址:www.nvidia.cn/Download/in…
一般情况下载
.run文件安装 - 进入命令行界面
按下
Ctrl+alt+f1登陆账户即可 - 关闭图形界面
sudo service lightdm stop
- 卸载原有驱动
sudo apt remove nvidia-* # --purge
如果使用以上命令卸载后运行nvidia-smi任然有显卡信息输出说明没有卸载成功,可使用以下方法卸载(使用runfile文件安装的驱动)
sudo sh ./NVIDIA-Linux-x86_64-xxx.run --uninstall
- 安装新驱动
sudo chmod a+x NVIDIA-Linux-x86_64-xxx.run
sudo ./NVIDIA-Linux-x86_64-xxx.run -no-x-check -no-nouveau-check -no-opengl-files
-no-x-check:安装驱动时关闭X服务 -no-nouveau-check:安装驱动时禁用nouveau -no-opengl-files:只安装驱动文件,不安装OpenGL文件
- 安装完成后,开启图形界面
sudo service lightdm start
- 验证是否安装成功
nvidia-smi
3. 循环登陆问题解决
如果安装好之后出现循环登陆问题,请从第二步开始,卸载刚安装过的驱动,重新安装一遍(安装命令记得后面加参数),
如果不成,多试几遍,多多踩坑
二、 安装cuda
1. 下载安装文件
developer.nvidia.com/cuda-toolki… 按照官方命令安装即可
sudo sh cuda_10.0.130_410.48_linux.run
注意:在安装过程中会询问一下东西,在问是否安装驱动时,如果前面已经安装过了,就选择否(最好是按前面的方法手动安装)
2. 设置环境变量,添加路径
sudo gedit ~/.bashrc
在文末添加一下信息
export PATH=/usr/local/cuda-10.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
激活环境
source ~/.bashrc
3. ubuntu 安装多版本cuda,及切换版本方法
文章传送门
三、安装cudnn
1.下载安装文件
cudnn下载 官网地址:developer.nvidia.com/rdp/cudnn-a…
注意:这里需要下载和cuda版本对应的文件
2. 安装cudnn
1. 解压下载文件
tar -xvf cudnn-10.0-linux-x64-xxx.tgz
2. 复制并给予权限
解压后是名为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
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
3. 查看cudnn版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2