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