解决Windows 10上torchvision报错“ImportError: DLL load failed: 找不到指定的模块”

1. 问题描述

在Windows 10上调用torchvision时报错,如下所示:

    from torchvision import _C
ImportError: DLL load failed: 找不到指定的模块。



2. 问题分析

Python环境中原先安装PyTorch、torchvision等Python包的时候,用的是pip安装的,可能存在某些编译或链接的问题。




3. 解决方法

S1. 改用conda重新安装PyTorch和torchvision包,这里以对应的CUDA版本为10.0为例,可以运行如下命令:

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

其他版本的CUDA,则上述命令可以改为如下形式:

conda install pytorch torchvision cudatoolkit=8.0 -c pytorch
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

如果没有GPU,只使用CPU,可以运行如下命令:

conda install pytorch-cpu torchvision-cpu -c pytorch

S2. 至此,问题得到解决。





参考资料:

https://pytorch.org/get-started/locally/

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章