ubuntu20.04安装pytorch-gpu cuda版本为11.7

安装 PyTorch GPU 版本需要先安装 CUDA 和 cuDNN。以下是在 Ubuntu 20.04 上安装 PyTorch GPU 版本的步骤：

1. 安装 CUDA 11.7

可以从 NVIDIA 官网下载 CUDA Toolkit 11.7 的安装包，选择适合自己系统的版本进行下载，下载地址为：https://developer.nvidia.com/cuda-downloads

下载完成后执行以下命令进行安装：

   sudo sh cuda_11.7.0_<version>_linux.run

安装过程中需要选择安装路径，一般选择默认路径即可。

2. 安装 cuDNN 8.2.2

前往 NVIDIA 官网下载 cuDNN 8.2.2 的安装包，下载地址为：https://developer.nvidia.com/cudnn-download-survey

注册一个 NVIDIA 开发者账号后就可以下载了。

下载完成后解压文件，并将 cuDNN 目录下的文件复制到 CUDA 的安装目录下的对应文件夹中。

   cd <path_to_cudnn_folder>
   sudo cp -P include/cudnn*.h /usr/local/cuda-11.7/include
   sudo cp -P lib64/libcudnn* /usr/local/cuda-11.7/lib64/
   sudo chmod a+r /usr/local/cuda-11.7/include/cudnn*.h /usr/local/cuda-11.7/lib64/libcudnn*

3. 安装 PyTorch

推荐使用 Anaconda 或 Miniconda 进行安装。

先创建一个新的环境：

   conda create --name pytorch_gpu python=3.9
   conda activate pytorch_gpu

然后安装 PyTorch：

   conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch -c conda-forge

安装完成后可以通过以下命令测试是否安装成功：

   python -c "import torch; print(torch.cuda.is_available())"

如果输出为 True，则说明安装成功。