深度学习项目GPU开发环境安装

注安装环境：ubuntu22.04, cuda 11.7, cudnn8.9

1.安装nvidia驱动

看可安装的Nvidia驱动版本：执行

ubuntu-drivers devices

安装合适版本的Nvidia驱动：

sudo apt-get install nvidia-driver-515

注意：合适的版本需要尝试（考虑与ubuntu、cuda版本的适配）
安装完成后，需要重启电脑：

sudo reboot

验证安装：

nvidia-smi

2.  安装cuda

NVIDIA CUDA驱动下载地址：https://developer.nvidia.com/cuda-toolkit-archive
选择合适的版本：（考虑ubuntu和pytorch版本的适配）

配置环境变量：

sudo gedit /etc/profile

文件末尾添加以下两条路径：

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

重启电脑：

sudo reboot

测试是否安装成功：

nvcc -V

如果显示没有nvcc命令，执行

sudo apt install nvidia-cuda-toolkit

3 .安装cudnn

下载合适的CuDNN版本：
去官网下载与CUDA搭配的cudnn版本。下载cudnn需要注册一个NIVDIA账号。
https://developer.nvidia.com/rdp/cudnn-download

下载后的文件：cudnn-local-repo-ubuntu2204-8.9.0.131_1.0-1_amd64.deb
执行

sudo dpkg -i cudnn-local-repo-ubuntu2204-8.9.0.131_1.0-1_amd64.deb

安装时提示：
 

The public cudnn-local-repo-ubuntu2204-8.9.0.131 GPG key does not appear to be installed.
To install the key

执行提示的指令。

sudo cp /var/cudnn-local-repo-ubuntu2204-8.9.0.131/cudnn-local-2063C34E-keyring.gpg
/usr/share/keyrings/