前言
提示:这里用的环境是Ubunt20.4.6 长期服务版 Ubuntu 20.04.6 LTS (Focal Fossa)
下载地址 https://releases.ubuntu.com/
一、CUDA安装
CUDA 安装可以参考 https://blog.csdn.net/takedachia/article/details/130375718
二、cuDNN安装
下载地址: https://developer.nvidia.com/cudnn-downloads、
根据系统版本选择然后执行以下命令就能安装了
# 不同的选项最后的命令可能不一样按官网显示的命令安装即可
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-keyring_1.1-1_all.deb
sudo dpkg -i cuda-keyring_1.1-1_all.deb
sudo apt-get update
sudo apt-get -y install cudnn