ubantu阿里下载pytorch

在Ubuntu系统上安装PyTorch，你可以按照以下步骤操作：

1. **更新包列表**:

   sudo apt-get update

2. **安装依赖库**:
PyTorch通常需要一些额外的开发工具和库，如CUDA和cuDNN (如果你计划在GPU上运行)。首先确保你的系统有支持CUDA的版本：

   sudo apt install nvidia-cuda-toolkit # 如果你有NVIDIA GPU

或者安装AMD ROCM如果使用AMD GPU:

   sudo apt install librocm-dev # 如果你有AMD GPU

3. **添加PyPI源**:
为了从非官方仓库安装最新版本的PyTorch，你需要添加一个PPA（Personal Package Archive）:

   sudo add-apt-repository ppa:pytorch/ppa

4. **更新并安装PyTorch**:

   sudo apt-get update
   sudo apt-get install python3-pip
   pip3 install torch torchvision torchaudio -f https://download.pytorch.org/whl/ubuntu focal.html

5. **安装特定版本**:
如果你需要指定版本，可以在安装命令后加上`--version=xx.x.x`，例如`pip3 install torch==1.9.0`.

6. **检查安装**:
安装完成后，你可以通过`python3 -c "import torch; print(torch.__version__)"`来确认PyTorch是否成功安装。