VS Code虚拟环境中解决PyTorch模块加载失败(WinError126)的方法
解决PyTorch导入时fbgemm.dll依赖错误的非Conda方案
1. 彻底清理虚拟环境里的PyTorch残留
直接重装没用是因为旧文件没清干净,按以下步骤操作:
- 激活你的虚拟环境:
.venv\Scripts\activate - 卸载所有PyTorch相关包:
pip uninstall -y torch torchvision torchaudio - 手动删除残留文件:找到
C:\Python\Test\.venv\Lib\site-packages下的torch文件夹,以及所有带torch-前缀的.dist-info文件夹,直接删除。
2. 重新安装适配的PyTorch版本
先安装CPU版本确认基础功能正常,之后再考虑GPU版本:
- 安装CPU版PyTorch:
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu - 若要安装GPU版,先查看自己的CUDA版本(命令行输入
nvcc --version),再对应安装,比如CUDA 12.1的话:
注意:Python3.11.9兼容当前官方提供的PyTorch版本,直接用上述命令即可。pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
3. 补装系统依赖库
WinError126错误大概率是缺失VC运行库,下载并安装**Microsoft Visual C Redistributable for Visual Studio 2022(x64版本)**,安装完成后重启电脑。
4. 验证修复效果
激活虚拟环境后运行测试代码:
import torch print(torch.__version__) # GPU版可以添加以下代码验证CUDA是否可用 print(torch.cuda.is_available())
内容的提问来源于stack exchange,提问作者Samuel
相关产品推荐
相关产品推荐

