0


ubuntu18.04安装cuda和cudnn

一、安装 cuda

1. 首先查看自身电脑最高支持的cuda版本为多少,在终端输入以下指令

nvidia-smi

可以看到我的最高支持 cuda 11.4 。

2. 打开英伟达官网 下载官方cuda

此处我下载的是cuda 11.0.3 版本,并选择相应配置,复制指令至终端下载

3. 运行官网安装指令

** 4. 依次选择continue和accept 点击确认**

** 5. 按回车取消Driver项的安装,然后回车选择Install**

** 6. 添加环境变量**
gedit ~/.bashrc打开文件,在文件结尾添加如下语句:

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

7. 保存、更新环境变量

source ~/.bashrc

8. 输入nvcc -V查看版本号

二、 安装 cudnn

1. 进入官网 ,寻找合适的 cudnn 版本,下载对应版本的cudnn 的 runtime developer 和samples

2. 下载得到四个文件

3. 安装上述软件包

tar -zxvf cudnn-11.0-linux-x64-v8.0.4.30.tgz
sudo cp -rf cuda/include/cudnn* /usr/local/cuda/include/
sudo cp -rf cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn*
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

sudo dpkg -i libcudnn8_8.0.4.30-1+cuda11.0_amd64.deb
sudo dpkg -i libcudnn8-dev_8.0.4.30-1+cuda11.0_amd64.deb
sudo dpkg -i libcudnn8-samples_8.0.4.30-1+cuda11.0_amd64.deb

** 4. 测试**

拷贝例程到用户目录

cp -r /usr/src/cudnn_samples_v8/ $HOME

5. 转到minist例程

cd  $HOME/cudnn_samples_v8/mnistCUDNN

6. 编译例程

sudo make clean && make

7. 运行例程

./mnistCUDNN

8. 如果cudnn被恰当的安装,将输出打印消息:Test passed!

9. cudnn 检测版本号

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2


本文转载自: https://blog.csdn.net/qq_43644413/article/details/124899366
版权归原作者 FAST-ROBOT 所有, 如有侵权,请联系我们删除。

“ubuntu18.04安装cuda和cudnn”的评论:

还没有评论