ubuntu 安装cuda cudnn环境

ubuntu 安装cuda cudnn环境_loong_XL的博客-CSDN博客

首先查看nvidia显卡驱动情况：nvidia-smi

参考：https://blog.csdn.net/weixin_45591044/article/details/104608506
https://littlede.blog.csdn.net/article/details/109407098

1、安装cuda
安装对于11.2的版本https://developer.nvidia.com/cuda-toolkit

wget https://developer.download.nvidia.com/compute/cuda/11.2.0/local_installers/cuda_11.2.0_460.27.04_linux.run

sudo sh cuda_11.2.0_460.27.04_linux.run
1
2
3

sudo vim ~/.bashrc

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
1
2
3
source ~/.bashrc

nvcc -V查看安装成功

2、cudnn安装
https://developer.nvidia.com/rdp/cudnn-archive 下载对应版本解压；可以微信授权登录获取下载链接

复制到一个目录下
tar -xzvf cudnn***v7.tgz

1
2
3
sudo cp cuda/include/* /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*
1
2
3
4
查看安装成功：

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

————————————————
版权声明：本文为CSDN博主「loong_XL」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/weixin_42357472/article/details/119991830