DELL G7系列,ubuntu16.04,本机显卡型号: GeForce GTX 1060。
主要步骤如下:
一,安装NVIDIA驱动。
1,本笔记本是F12进入BIOS,将secure boot设置为disable。
2,通过下面命令确保电脑上之前的nvidia驱动已卸载干净(可选)。
sudo apt-get remove --purge nvidia*
3,通过下面命令卸载之前下载的驱动安装包(runfile格式)(可选)。
sudo sh /home/xxx/驱动名.run --uninstall
4,屏蔽开源驱动nouveau,这里选择blacklist,首先创建新文件。
sudo vim /etc/modprobe.d/blacklist-nouveau.conf
内容如下:
blacklist nouveauoptions nouveau modeset=0
再通过如下命令更新:
sudo update-initramfs -u
通过如下命令,nouveau驱动是否被禁用(无输出则被禁用)。
lsmod | grep nouveau
5,打开终端加入,加入官方ppa源。
sudo add-apt-repository ppa:graphics-drivers/ppa
6,刷新软件库。
sudo apt-get update
7,查看推荐的显卡驱动,选择驱动后边有推荐(recommended)
字符的。
ubuntu-drivers devices
8,通过下面命令安装推荐的显卡驱动,并在完成后重启。
sudo apt-get install nvidia-430 nvidia-settings nvidia-primereboot
9,检查(确保两个命令都有正常输出)。
下面命令显示显卡的一些内容信息。
nvidia-smi
设置显卡。
nvidia-settings
二,安装cuda和cudnn
1,cuda官网下载,/cuda-10.0-download-archive,选择runfile版本。
2,将run文件拷贝到HOME下面,在shell下,HOME目录里执行如下命令。将cuda_10.0.130_410.48_linux.run重命名为cuda.run。
sudo sh cuda.run
这里会显示用户认证信息,按回车键加载到100%。先accept,INSTALL NVIDIA ACCELERATED GRAPHICS DRIVER............选择no。
后面就一路yes和回车即可。
3,设置环境变量。
sudo gedit /etc/profile
并在文件末尾添加如下两行。
export PATH=/usr/local/cuda-10.0/bin${PATH:+:${PATH}}export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
并执行如下命令,保证了环境变量的永久配置。
source /etc/profilereboot
4,验证驱动版本和CUDA版本。
cat /proc/driver/nvidia/version
nvcc -V
5,安装cudnn
cuDNN Archive
需要先注册NVIDIA帐号,并通过如下命令解压(在下载好的目录里)。
tar -xvf cudnn-10.0-linux-x64-v7.3.1.20.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.hsudo chmod a+r /usr/local/cuda/lib64/libcudnn*
并查看cudnn版本。
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
参考链接:
Ubuntu16.04上CUDA_10.0及cuDNN的安装_lihe的博客-CSDN博客