Ubuntu 16.04 安装NVIDIA显卡驱动及CUDA、CuDnn(详细)

一、安装NVIDIA显卡驱动

1、首先卸载原有NVIDIA驱动:

sudo apt-get purge nvidia*

2、禁用nouveau驱动

      在终端输入:

sudo gedit /etc/modprobe.d/blacklist-nouveau.conf

     添加: 

blacklist nouveau
options nouveau modeset=0
     一定记着更新:
sudo update -initramfs -u   

     更新完成后重启系统

sudo reboot
     重启后输入:
lsmod | grep nouveau

     如果没有输出结果则正确。

3、下载NVIDIA驱动

     http://www.nvidia.cn/Download/index.aspx?lang=cn

选择对应的版本下载

4、在终端输入

sudo service lightdm stop

     按Ctrl+Alt+F1,切换到tty1,需要再次输入Ubuntu用户名和密码。

5、安装驱动

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-opengl-files也就是不安装opengl,这个是必须要的,否则可能出现循环登陆的问题,–no-x-check 和–no-nouveau-check 是安装驱动时不检查X服务和nouvea,这两句不是必要的。

    需要选择时按默认即可。

6、在终端输入,返回。

sudo service lightdm start

7、检查NVIDIA,终端输入

nvidia-smi

二、安装CUDA

1、进入官网选择对应版本下载:

https://developer.nvidia.com/cuda-downloads

在终端输入:

sudo bash ***.run

注意,此时遇到询问是否安装驱动的选项,选择NO,不再安装驱动

2、添加到环境变量

 (1)终端输入

sudo gedit ~/.bashrc

   添加:

export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} 

 (2)终端输入

sudo gedit /etc/profile

   添加:

export PATH=/usr/local/cuda/bin:$PATH
export PATH=/usr/local/cuda/bin:$PATH
 (3)终端输入
sudo gedit /etc/ld.so.conf.d/cuda.conf

    添加:

/usr/local/cuda/lib64
  (4)终端输入
sudo ldconfig

3、简单验证:

nvcc -V

三、配置CuDnn

1、官网下载;

https://developer.nvidia.com/rdp/cudnn-download

需要先注册一个账号,再选择对应版本下载

2、解压

3、移动文件

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/

4、重新生成软链接(具体内容根据实际情况修改)

sudo rm -rf libcudnn.so libcudnn.so.7
sudo ln -s libcudnn.so.7.0.4 libcudnn.so.7
sudo ln -s libcudnn.so.7 libcudnn.so

5、完成

阅读更多
想对作者说点什么?

博主推荐

换一批

没有更多推荐了,返回首页