ubuntu使用pip安装pytorch

在Ubuntu上使用pip安装PyTorch可以按照以下步骤进行：

1. 打开终端并输入以下命令以更新pip：

pip install --upgrade pip

2. 确认你的CUDA版本，例如11.3。

3. 运行以下命令以安装PyTorch：

pip install -i https://mirrors.aliyun.com/pypi/simple/ torch==1.11.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html

其中，1.11.0是你要安装的PyTorch版本，cu113是你的CUDA版本。

4. 运行以下命令以安装torchvision：

pip install torchvision