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

ubuntu18.04安装cuda10.1和cudnn7

程序员文章站 2022-03-03 14:47:42
...

参考:
安装
安装

安装cuda10.1

第一步

cuda官网选择适合自己电脑的版本下载

第二步

在终端输入在第一步中选择好版本后官网给出的安装命令。

第三步

由于显卡驱动是之前就单独安装好的,所以在这里安装cuda时注意不要再安装驱动driver了。除了驱动不装,其他的都都安装。一路指令顺序执行下来。

第四步

环境变量的配置。打开 ~/.bashrc 文件补充代码。
在终端输入:

sudo gedit ~/.bashrc

在打开的文件底端补充:

export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

输入完成后,保存并且关闭文档,终端界面上的警告可以忽略。然后再在终端输入:

source ~/.bashrc

让配置生效

第五步

创建链接文件,终端输入:

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

在打开的文件中添加语句:

/usr/local/cuda/lib64

文档保存关闭后终端输入:

sudo ldconfig
第六步

终端输入

nvcc  -V

查看cuda是否安装成功

安装cudnn

第一步

官网下载安装包,需要注册登录才能下载,选择适合的版本,最后选择的是cudnn for linux

第二步

下载完成后将压缩包解压缩,进入该文件夹。打开终端,输入

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*
第三步

在终端查看cudnn版本

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
相关标签: 安装