准备工作
1、开机进ubuntu设计界面按e禁用nouveau驱动操作流程
2、去nvidia官网下载CUDA安装包下载传送门
3、下载CUDNN我是传送门
4、安装gcc 6 或者降级降级
sudo apt-get install gcc-6
5、安装make
sudo apt-get install make
安装CUDA
1、给run安装包可执行权限
chmod 777 cudaxxx1.run #有多少个执行多少次
chmod 777 cudaxxx2.run
2、执行
sudo ./cudaxxx1.run #有多少个执行多少次
sudo ./cudaxxx2.run
安装时一般都选accept default y
只需注意若是已经安装高于cuda推荐版本的nvidia驱动时,在选择是否安装驱动是选n,反之y
3、设置环境变量
sudo gedit ~/.bashrc
在最后加入
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
4、验证cuda是否安装成功
重启终端输入
nvcc -V
出现如下
安装CUDNN
1、解压
tar -xvf cudnnxxx.tar
2、复制
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、验证是否安装成功
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
出现如下,即成功
永久禁用nouveau
sudo gedit /etc/modprobe.d/blacklist.conf
输入密码后在最后一行加上: blacklist nouveau
是将Ubuntu自带的显卡驱动加入黑名单
Ctrl +S保存后注意此时还需执行以下命令使禁用 nouveau 真正生效终端输入 :
sudo update-initramfs -u
重启电脑
附录
1、查看已装nvidia驱动的版本
sudo dpkg --list | grep nvidia-*
2、卸载nvidia驱动或者cuda
-------------nvidia
sudo apt-get --purge remove nvidia-*
# 也有的说用下面的命令进行卸载,我个人均有尝试,无碍
sudo apt-get --purge remove "*nvidia*"
-------------cuda
sudo apt autoremove cuda
# 若上一步删不干净,接着使用:
sudo apt --purge remove "*cublas*" "cuda*"