ubuntu20.04 本地安装pytorch

在Ubuntu 20.04上安装PyTorch，您需要先确保已经安装了必要的依赖包，如Python、CUDA和cuDNN。以下是安装步骤：

1. **更新系统并安装基本工具**:

   sudo apt update && sudo apt install -y python3-pip software-properties-common

2. **添加PyPI官方镜像** (如果你的网络连接不稳定，可以考虑替换为中国源):

   curl https://bootstrap.pypa.io/get-pip.py | python3
   pip install torch torchvision --index-url https://mirrors.aliyun.com/pypi/simple/

3. **检查CUDA版本**:
如果你想使用GPU加速，首先确保已安装CUDA和cuDNN。运行以下命令（这里假设你要安装CUDA 11.x）：

   sudo add-apt-repository ppa:ptf/ptf
   sudo apt-get update
   sudo apt-get install cuda cudnn

然后查看CUDA版本和选择对应的PyTorch版本，例如对于CUDA 11.0：

   pip install torch==1.7.1+cu110 torchvision==0.8.1+cu110 torchaudio===0.7.1

4. **安装CPU版本** (如果不需要GPU支持)：

   pip install torch torchvision torchaudio

5. **验证安装**:
运行 `python3 -c "import torch; print(torch.__version__)"` 来确认PyTorch是否成功安装。