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