Ubuntu16.04安装CUDA9.1、cuDNN7详细教程(在xfce驱动是9.0)

参考链接:https://blog.csdn.net/Jonms/article/details/79318566

安装CUDA 9.1 

到NVIDIA官网中,找到CUDA9.1并下载 

找到下载文件的路径,键入下面的命令安装:

$sudo sh cuda_9.1.85_387.26_linux.run

单击回车,直到提示“是否为NVIDIA安装驱动?” 

选择否,因为已经安装好驱动程序,其他都是默认。 

最后你会看到cuda驱动、sample、tookit已经安装成功,但是缺少一些库。

添加这些库

$ sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev

重新启动图形化界面

$ sudo service lightdm start

接着,设置环境变量。 

终端中输入

$ sudo vim /etc/profile

在打开的文件末尾,添加以下两行。 

64位系统:

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

保存文件,并重启。因为source /etc/profile是临时生效,重启电脑才是永久生效。

重启电脑,检查上述的环境变量是否设置成功。 

a.验证驱动版本

$ cat /proc/driver/nvidia/version

重启电脑,检查上述的环境变量是否设置成功。 

a.验证驱动版本

$ cat /proc/driver/nvidia/version

b.验证CUDA Toolkit

$ nvcc -V

cuDNN的安装是建立在我们成功安装cuda的基础上的,cuDNN同样需要我们去NVIDIA的官网下载适合cuda版本的deb文件或tgz文件。 
这里我以tgz文件的安装为例(cudnn-9.1-linux-x64-v7.tgz), 

首先解压tgz文件

$ tar -xzvf cudnn-9.1-linux-x64-v7.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.h /usr/local/cuda/lib64/libcudnn*

猜你喜欢

转载自blog.csdn.net/qq_35292447/article/details/80094020
今日推荐