ubuntu 安装cuda cudnn环境

首先查看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

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

source ~/.bashrc

nvcc -V查看安装成功

2、cudnn安装

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

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

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*

查看安装成功：

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