Ubuntu20.04环境下Pytorch安装教程

以下是在 Ubuntu20.04 环境下安装 PyTorch 的步骤：

1. 安装 Python3 和 pip3：

   sudo apt-get update
   sudo apt-get install python3 python3-pip

2. 安装 PyTorch：

   pip3 install torch torchvision

这个命令会安装最新的 CPU 版本的 PyTorch。如果你想使用 GPU 版本，可以使用以下命令：

   pip3 install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html

这个命令会安装 CUDA 11.1 版本的 PyTorch。

3. 测试安装是否成功：

   python3
   >>> import torch
   >>> print(torch.__version__)

如果能够输出 PyTorch 的版本号，那么说明安装成功。

希望这个教程可以帮助到你。