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