前提
- 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配置到环境变量即可。
再次重新编译完成。