ubuntu20.04安装pytorch gpu

安装 PyTorch GPU 版本需要先安装 CUDA 和 cuDNN，然后再使用 pip 安装 PyTorch。以下是具体步骤：

1. 安装 CUDA 和 cuDNN

首先需要安装 CUDA 和 cuDNN，可以参考 NVIDIA 的官方文档进行安装。需要注意的是，PyTorch 版本和 CUDA、cuDNN 版本有对应关系，需要根据 PyTorch 官方文档中的说明选择对应的版本。

2. 创建虚拟环境

推荐使用 Anaconda 创建虚拟环境，可以避免与系统环境冲突。可以使用以下命令创建一个名为 pytorch 的虚拟环境：

   conda create --name pytorch python=3.8

3. 激活虚拟环境

创建完虚拟环境后，需要激活虚拟环境才能在其中安装 PyTorch。可以使用以下命令激活 pytorch 环境：

   conda activate pytorch

4. 安装 PyTorch

在激活虚拟环境后，可以使用 pip 安装 PyTorch。可以使用以下命令安装最新版本的 PyTorch：

   pip install torch torchvision torchaudio

如果需要安装指定版本的 PyTorch，可以在命令中指定版本号，例如：

   pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/cu111/torch_stable.html

需要注意的是，PyTorch 版本和 CUDA、cuDNN 版本有对应关系，需要根据 PyTorch 官方文档中的说明选择对应的版本。