参考链接: Ubuntu16.04下安装cuda和cudnn的三种方法(亲测全部有效)
Ubuntu16.04LTS安装Nvidia显卡驱动+cuda8.0+cudnn
一、安装CUDA
1.下载cuda
2.禁用ubuntu自带的nouveau
终端中运行:$ lsmod | grep nouveau,如果有输出则代表nouveau正在运行
禁用nouveau
输入命令:$ sudo vim /etc/modprobe.d/blacklist-nouveau.conf
在文件中输入:
blacklist nouveau
options nouveau modeset=0
保存退出终端执行以下命令
$ sudo update-initramfs –u
重启进入高级启动root模式
运行CUDA安装包
然后可以自动安装驱动也可以用命令行选择系统匹配的驱动
sudo apt-get install nvidia-390
ubuntu 16.04 安装cuda 8 出现的错误:Missing recommended library: libGLU.so……
sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev
3.设置环境变量
重启进入系统后设置环境变量
终端中输入 $ sudo gedit /etc/profile
在打开的文件末尾,添加以下两行。
64位系统:
export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64:${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
32位系统:
export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib\
${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
保存文件,并重启。因为source /etc/profile是临时生效,重启电脑永久生效。
4.检查是否安装成功
a、 验证驱动版本
$ cat /proc/driver/nvidia/version
结果显示
NVRM version: NVIDIA UNIX x86_64 Kernel Module 384.81 Sat Sep 2 02:43:11 PDT 2017
GCC version: gcc version 5.4.0 20160609 (Ubuntu 5.4.0-6ubuntu1~16.04.5)
b、 验证CUDA Toolkit
$ nvcc -V 会输出CUDA的版本信息
如果是这样的:
The program 'nvcc' is currently not installed. You can install it by typing:
sudo apt-get install nvidia-cuda-toolkit
则说明环境变量没有安装好
二、安装CUDNN
1.下载cudnn-10.0-linux-x64-v6.0.tgz以后,解压到当前目录下 下载链接
2.
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.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
3.检查是否安装成功
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2