欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

ubuntu16.04驱动+cuda9.0+cudnn7.0

程序员文章站 2024-01-15 23:52:46
...

Ubuntu16.04(系统不同可能会略有不同)

1.驱动安装

千万别用那种禁掉原显卡那种的方法,全是坑,系统容易崩!!!

安装依赖项:

sudo apt-get update

sudo apt-get install dkms build-essential linux-headers-generic

终端输入

sudo apt-get install nvidia-

开始按tab键补全,在里面找到型号

ubuntu16.04驱动+cuda9.0+cudnn7.0

这里我选的384,可根据自己的型号,把384换掉,输入:

sudo apt-get install nvidia-384 nvidia-settings nvidia-prime
检查是否安装成功

终端输入:

nvidia-smi

出现类似如下:

ubuntu16.04驱动+cuda9.0+cudnn7.0

终端输入:

nvidia-settings
出现类似如下:

ubuntu16.04驱动+cuda9.0+cudnn7.0

表示已成功安装驱动。

2.Cuda9.0

官网链接:

https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=runfilelocal

安装相关依赖项:

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

修改文件权限:

sudo chmod 777 cuda_9.0.176_384.81_linux.run
回车之后按空格阅读完之后,按照提示,依次输入

accept->n->y->回车->y->y->回车,等待安装

 配置环境变量,在文件末尾添加两行:

sudo gedit  /etc/profile

export  PATH=/usr/local/cuda-9.0/bin:$PATH
export  LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64$LD_LIBRARY_PATH
重启电脑

测试CUDA的samples

cd /usr/local/cuda-9.0/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery

出现类似下图

ubuntu16.04驱动+cuda9.0+cudnn7.0

安装一些依赖库

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

添加环境变量

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}}

设置环境变量和动态链接库

sudo gedit /etc/profile

export PATH=/usr/local/cuda/bin:$PATH

创建链接文件/etc/ld.so.conf.d/cuda.conf

sudo gedit /etc/ld.so.conf.d/cuda.conf

在/etc/ld.so.conf.d/cuda.conf写入:

/usr/local/cuda/lib64

执行:

sudo ldconfig

3.Cudnn7.0

下载Cudnn7.0压缩包,官网需要注册才能下载

官网地址:

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

或者:

https://download.csdn.net/download/daydayjump/10051196?locationNum=1&fps=1

下载后,解压压缩包:

tar -zxvf cudnn-9.0-linux-x64-v7.0.tgz
进入解压后的cuda文件夹,进行拷贝:
cd cuda

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

进入/usr/local/cuda/lib64/,更新网络链接(如果不是cudnn7.0的话需要更改下面的后缀

cd /usr/local/cuda/lib64/

sudo chmod +r libcudnn.so.7.0.3
sudo ln -sf libcudnn.so.7.0.3. libcudnn.so.7
sudo ln -sf libcudnn.so.7 libcudnn.so
sudo ldconfig

验证是否安装成功

nvcc -V
如果出现类似下图,即安装成功

ubuntu16.04驱动+cuda9.0+cudnn7.0



参考:

https://www.jianshu.com/p/b313625fbe61

https://blog.csdn.net/fdqw_sph/article/details/78745375