添加链接
link之家
链接快照平台
  • 输入网页链接,自动生成快照
  • 标签化管理网页链接
首发于 技术杂谈
Ubuntu安装CUDA与CUDNN完全手册

Ubuntu安装CUDA与CUDNN完全手册

系统:Ubuntu 16.04 64bit

硬件:Geforce 940MX(my computer)

安装Nvidia的Linux显卡驱动

1. 到Nvidia官网根据自己的显卡型号下载 *.run文件: geforce.cn/drivers

2.由于Ubuntu系统一般用使用开源的显卡驱动nouveau,因此需首先将其屏蔽

sudo gedit /etc/modprobe.d/blacklist.conf,添加下述内容

blacklist vga16fb

blacklist nouveau

blacklist rivafb

blacklist nvidiafb

blacklist rivatv

3. 若以前安装过Nvidia驱动,则需先将其删除

sudo apt-get --purge remove nvidia-*

sudo apt-get --purge remove xserver-xorg-video-nouveau

4. sudo update-initramfs -u,然后重启电脑(在微星GTX1070上安装时没有加上这条命令行,导致安装总是出现 unable to load the kernel module 'nvidia.ko'问题; 但是用dell的940mx没有加上这条指令仍然好用。因此最好加上这条指令)

5.Ctrl+Alt+F1/F4,进入无界面命令行模式,然后使用下面命令彻底关闭图形界面

sudo service lightdm stop

6.安装Nvidia驱动

sudo ./Nvidia-Linux-x86_64-xxx.xx.run -no-opengl-files -no-x-check -no-nouveau-check

开始选择Accept

报错"The distribution-provided pre-install script failed"不用理会,选Yes

提示是否使用Nvidia的xconfig文件时选Yes

其余选项都选Yes,直到安装成功

Note: -no-opengl-files : 不安装OpenGL,非常重要,否则会出现系统循环登入问题

-no-x-check : 安装驱动时不检查x服务(可以不加)

-no-nouveau-check : 安装驱动时不检查nouveau(可以不加)

7.重新启动图形界面

sudo service lightdm start

8.测试是否安装成功

sudo nvidia-smi,看是否显示Nvidia显卡信息,则为安装正确

9.重启可能遇到问题

问题:重启后显示分辨率变低,但是能登入

解决:重新安装一遍显卡驱动即可

10.卸载Nvidia驱动

命令行界面下: $ sudo ./Nvidia-Linux-x86_64-xxx.xx.run uninstall


安装CUDA

1.CUDA文件下载

根据操作系统版本,下载对应的CUDA文件: developer.nvidia.com/cu

过往版本的cuda: developer.nvidia.com/cu

2.命令行安装

sudo ./cuda_8.0.61_375.26_linux.run.26_linux-run,安装过程中基本选择默认路径与YES,但提到是否安装

Nvidia驱动时选NO

3.配置环境变量

在~/.bashrc中添加下述环境变量

export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}

export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

4.验证

cd /usr/local/cuda/samples/1_Utilities/deviceQuery

make

sudo ./deviceQuery

5.卸载

在/usr/local/cuda-8.0/bin下运行uninstall script即可


安装Cudnn

1.下载文件

根据CUDA的版本,到Nvidia官网下载对应的cudnn驱动文件: developer.nvidia.com/rd

2.安装

若下载得到的文件是以为solitairetheme8后缀的,则先要进行下面的步骤

$ cp cudnn-8.0-linux-x64-v5.1.solitairetheme8 cudnn-8.0-linux-x64-v5.1.tgz

$ tar -xvf cudnn-8.0-linux-x64-v5.1.tgz

否则下载后解压,进入include文件夹,执行:sudo cp cudnn.h /usr/local/cuda/include/

进入lib64文件夹,执行:sudo cp lib* /usr/local/cuda/lib64/

执行软链接:

cd /usr/local/cuda/lib64/

sudo rm -rf libcudnn.so libcudnn.so.6 #删除原有动态文件

sudo ln -s libcudnn.so.6.0.21 libcudnn.so.6 #生成软链接

sudo ln -s libcudnn.so.6 libcudnn.so #生成软链接


本文整理自: cnblogs.com/wmxfd/p/ins

发布于 2019-06-28 11:36

文章被以下专栏收录