禁用Ubuntu的nouveau:
sudo gedit /etc/modprobe.d/blacklist-nouveau.conf
在文件中输入以下下内容:
blacklist nouveau
options nouveau modeset=0
sudo update-initramfs -u
重启
lsmod | grep nouveau
若无内容输出,则禁用成功。
ctrl+alt+F1进入终端界面,关闭图形化界面
sudo service lightdm stop
安装cuda,按q可跳过说明
sudo sh cuda_7.5.18_linux.run
遇到提示是否安装openGL ,选择no(如果你的电脑跟我一样是双显,且主显是非NVIDIA的GPU在工作需要选择no,否则可以yes),其他都选择yes或者默认即可。(如果您的电脑是双显卡且在这一步选择了yes,那么你极有可能安装完CUDA之后,重启图形化界面后遇到登录界面循环问题:输入密码后又跳回密码输入界面。
重新启动图形化界面
sudo service lightdm start
Alt + ctrl +F7切回图像化界面。
如果出现重复登陆等异常情况,卸载驱动重新安装
sudo /usr/local/cuda-8.0/bin/uninstall_cuda_8.0.pl
sudo /usr/bin/nvidia-uninstall
查看安装情况
ls /dev/nvidia*
若结果显示以下下或类似的信息,则安装成功
/dev/nvidia0 /dev/nvidiactl /dev/nvidia-uvm
若显示以下
ls: cannot access/dev/nvidia*: No such file or directory
或是这样的,只出现
/dev/nvidia0 /dev/nvidiactl
执行
sudo gedit /etc/rc.local
替换文本
#!/bin/bash
/sbin/modprobe nvidia
if [ "$?" -eq 0 ]; then
# Count the number of NVIDIA controllers found.
NVDEVS=`lspci | grep -i NVIDIA`
N3D=`echo "$NVDEVS" | grep "3D controller" | wc -l`
NVGA=`echo "$NVDEVS" | grep "VGA compatible controller" | wc -l`
N=`expr $N3D + $NVGA - 1`
for i in `seq 0 $N`; do
mknod -m 666 /dev/nvidia$i c 195 $i
done
mknod -m 666 /dev/nvidiactl c 195 255
else
exit 1
fi
/sbin/modprobe nvidia-uvm
if [ "$?" -eq 0 ]; then
# Find out the major device number used by the nvidia-uvm driver
D=`grep nvidia-uvm /proc/devices | awk '{print $1}'`
mknod -m 666 /dev/nvidia-uvm c $D 0
else
exit 1
fi
exit 0
重新查看状态
ls /dev/nvidia*
显示如下则成功
/dev/nvidia0 /dev/nvidiactl /dev/nvidia-uvm
设置环境变量
sudo gedit /etc/profile
在打开的文件末尾,添加以下两行
export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda
测试是否成功,显示信息则成功
source /etc/profile
cat /proc/driver/nvidia/version
nvcc -V
重启电脑。
安装cudnn,注意版本
sudo dpkg -i libcudnn7_7.5.0.56-1+cuda10.0_amd64.deb