DigitalOcean Ubuntu实例安装torch后出现ModuleNotFoundError问题

问题详情 #

在DigitalOcean的Ubuntu Droplet上安装PyTorch，执行以下步骤创建虚拟环境：

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安装显示成功，但执行import torch时报错：

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

原因分析 #

1. 旧版本PyTorch已下架：PyTorch 1.7.1属于较旧版本，官方已从PyPI索引中移除，因此无法通过指定版本号安装。
2. Python环境不匹配：安装成功但无法import，核心原因是安装时使用的pip和运行时调用的Python不属于同一个虚拟环境——可能是激活虚拟环境后，pip仍指向系统全局的pip，或者运行python时调用了系统默认的Python解释器，而非虚拟环境内的版本。

解决步骤 #

1. 确认虚拟环境处于激活状态 #

确保命令行前缀显示(pytorch)，如果没有，重新激活：

source ~/pytorch/pytorch/bin/activate

2. 安装兼容的PyTorch版本 #

选择官方索引中存在的CPU版本，例如2.1.0+cpu：

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

3. 验证安装（使用虚拟环境内的Python） #

直接执行以下命令验证，避免手动进入Python时选错解释器：

python -c "import torch; print(torch.__version__)"

如果输出版本号（如2.1.0+cpu），说明安装成功。

4. 环境路径排查（若仍失败） #

检查pip和python的路径是否属于虚拟环境：

which pip
which python

正常情况下，路径应指向~/pytorch/pytorch/bin/目录下的对应文件。如果不是，说明虚拟环境激活异常，可尝试删除现有虚拟环境后重新创建：

rm -rf ~/pytorch/pytorch
python3 -m venv ~/pytorch/pytorch
source ~/pytorch/pytorch/bin/activate

之后重新执行安装命令。

内容的提问来源于stack exchange，提问作者user3574603