安装detectron2找不到torch模块

0 阅读1分钟

前提

  • windows11
  • python 3.12.0
  • visual studio installer 中下载好C++ CMake工具
  • wheel 0.45.1
  • cython 3.1.1
  • torch 2.6.0+cu124 (显卡不一样版本也就不一样)

场景

我遇到的场景是,以可编辑模式去安装发生torch模块找不到!

ModuleNotFoundError: No module named 'torch'

执行一下验证代码是否有正确输出

python -c "import torch; print(torch.__file__)"
//输出的是你的torch模块的实际路径
// C:\admin\AppData\Local\Programs\Python\Python312\Lib\site-packages\torch\__init__.py

解决

保证能够正常输出后,查看你的setuptools 版本是否高于75.8.0

pip show setuptools

如果高于则可以尝试降级解决(这只适用于我)

pip install setuptools==75.8

再执行编译安装

编译后找不到cl执行文件

//报错信息
Python\Python312\Lib\site-packages\torch\utils\cpp_extension.py:414: UserWarning: Error checking compiler version for cl

参考下图,根据安装地址找到对应的cl安装目录

将D:\App\Microsoft Visual Studio\2022\BuildTools\VC\Tools\MSVC\14.43.34808\bin\Hostx64\x64配置到环境变量即可。 image.png

再次重新编译完成。