解决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/