Python3 + Digital Ocean：没有名为“torch”的模块

Question

我正在尝试在 Digital Ocean Ubuntu Droplet 上安装 torch。我正在关注这个：

mkdir ~/pytorch
mkdir ~/pytorch/assets
cd ~/pytorch
python3 -m venv pytorch
source pytorch/bin/activate

运行命令

pip install torch==1.7.1+cpu torchvision==0.8.2+cpu -f https://download.pytorch.org/whl/torch_stable.html

会导致错误：

ERROR: Could not find a version that satisfies the requirement torch==1.7.1+cpu (from versions: 1.13.0, 1.13.0+cpu, 1.13.0+cu116, 1.13.0+cu117, 1.13.0+cu117.with.pypi.cudnn, 1.13.1, 1.13.1+cpu, 1.13.1+cu116, 1.13.1+cu117, 1.13.1+cu117.with.pypi.cudnn, 2.0.0, 2.0.0+cpu, 2.0.0+cpu.cxx11.abi, 2.0.0+cu117, 2.0.0+cu117.with.pypi.cudnn, 2.0.0+cu118, 2.0.1, 2.0.1+cpu, 2.0.1+cpu.cxx11.abi, 2.0.1+cu117, 2.0.1+cu117.with.pypi.cudnn, 2.0.1+cu118, 2.0.1+rocm5.3, 2.0.1+rocm5.4.2, 2.1.0, 2.1.0+cpu, 2.1.0+cpu.cxx11.abi, 2.1.0+cu118, 2.1.0+cu121, 2.1.0+cu121.with.pypi.cudnn, 2.1.0+rocm5.5, 2.1.0+rocm5.6)
ERROR: No matching distribution found for torch==1.7.1+cpu

所以我尝试了以下两种方法（似乎运行成功）：

pip3 install torch torchvision
pip install torch torchvision

但是，当我尝试在 python 中导入 torch 时，出现错误：

>>> import torch
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ModuleNotFoundError: No module named 'torch'

为什么python找不到torch？我需要做什么才能确保我可以使用该模块？

Answer 1

你可以尝试一下：

pip install torch==2.0.1+cpu torchvision torchaudio -f
https://download.pytorch.org/whl/torch/

你的轮子将是：

torch-2.1.0+cpu-cp311-cp311-win_amd64.whl