Ubuntu 16.04 上安装 CUDA 9.0 + cudnn7.0 详细教程

最新推荐文章于 2024-08-14 09:08:25 发布

ei1994

最新推荐文章于 2024-08-14 09:08:25 发布

阅读量1.3k

点赞数

分类专栏： python

python 专栏收录该内容

24 篇文章 0 订阅

订阅专栏

参考文献：
https://blog.csdn.net/QLULIBIN/article/details/78714596
https://blog.csdn.net/fdqw_sph/article/details/78745375#commentBox
超详细安装教程：http://blog.51cto.com/moerjinrong/2175362
还有这个https://cloud.tencent.com/developer/article/1086781
再加一个https://www.jianshu.com/p/591ba8c23959

CUDA安装时会自动安装nvidia显卡驱动，这会比自己安装省很多问题（别问我是怎么知道的）。之后如果哪一个崩溃了，就单独按那个就好。

找到下载文件的路径

  sudo sh cuda_9.0.176_384.81_linux.run

单击回车 ，一路往下运行，直到提示“是否为NVIDIA安装驱动nvidia-384？”，选择是 ，因为已经安装好驱动程序了，其他的全都是默认，不过要记住安装位置，默认是安装在/usr/local/cuda文件夹下。
配置环境变量，运行如下命令打开profile文件

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

保存，然后重启电脑

sudo reboot

环境变量配置
安装完毕后，再声明一下环境变量，并将其写入到 ~/.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

保存之后，创建链接文件：
sudo gedit /etc/ld.so.conf.d/cuda.conf
在打开的文件中添加如下语句
/usr/local/cuda/lib64
然后执行
sudo ldconfig
使链接立即生效。

5.安装cudnn

继续Ctrl+alt+F1进入的字符界面
进入解压后的cudnn-9.0-linux-x64-v7.0.tgz文件cuda，在终端执行下面的指令安装：

tar -zxvf cudnn-9.0-linux-x64-v7.0.tgz
cd cuda
sudo cp lib64/lib* /usr/local/cuda/lib64/
sudo cp include/cudnn.h /usr/local/cuda/include/
然后更新网络连接：

cd /usr/local/cuda/lib64/
sudo chmod +r libcudnn.so.7.0.3 # 自己查看.so的版本
sudo ln -sf libcudnn.so.7.0.3. libcudnn.so.7
sudo ln -sf libcudnn.so.7 libcudnn.so
sudo ldconfig
重新启动图形化界面
sudo service lightdm start
再Ctrl+alt+F7退出Text Mode。