Ubuntu安装Detectron2报错：已装PyTorch却提示找不到torch模块

问题现象 #

在Ubuntu系统的虚拟环境detectron2_test中，已通过pip安装PyTorch（2.4.1+cu118）和torchvision（0.19.1+cu118），且pip、python、python3均指向虚拟环境路径，但执行pip install 'git+https://github.com/facebookresearch/detectron2.git'时，报错ModuleNotFoundError: No module named 'torch'。

解决建议 #

• 强制使用虚拟环境的Python执行pip安装
  直接调用虚拟环境内的Python运行pip，避免路径混淆导致调用系统级pip：

  python -m pip install 'git+https://github.com/facebookresearch/detectron2.git'
  

• 验证当前环境中torch的可导入性
  在虚拟环境中执行以下命令确认torch正常：

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

  如果此处报错，说明torch安装不完整，重新安装对应版本：

  pip install torch==2.4.1+cu118 torchvision==0.19.1+cu118 --index-url https://download.pytorch.org/whl/cu118
  

• 重新激活虚拟环境
  即使环境前缀显示正常，也可能存在激活状态异常，重新激活后再尝试安装：

  source /home/ubuntu/detectron2_test/bin/activate
  

• 本地克隆仓库后安装
  跳过git直接安装的路径问题，克隆仓库到本地后进行源码安装：

  git clone https://github.com/facebookresearch/detectron2.git
  cd detectron2
  pip install -e .
  

• 检查Python版本兼容性
  Detectron2对Python3.12的支持可能不完善，建议切换到Python3.10或3.11版本，重新创建虚拟环境后再执行安装流程。

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