lsmod | grep nouveau 查看nouveau是否使用,无输出则意味着被禁用或没有
有则执行sudo vim /etc/modprobe.d/blacklist-nouveau.conf, 输入
blacklist nouveau
options nouveau modeset=0
保存退出
在终端执行sudo update-initramfs -u
重启电脑
cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery
sudo make
sudo ./deviceQuery
保证编译没有error,执行deviceQuery输出信息的最后类似以下:
deviceQuery, CUDA Driver = CUDART, CUDA Driver Version = 8.0, CUDA Runtime Version = 8.0, NumDevs = 1, Device0 = GeForce GTX 960
Result = PASS
6.cudnn安装
tar xvzf cudnn-8.0-linux-x64-v5.1.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/*.* /usr/local/cuda/lib64 -a
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*