在使用PyTorch时出现错误:“from torch._C import * ImportError: DLL load failed: 找不到指定的模块”,其实是CUDA版本与pytoch版本不对应导致。
在使用导入torchvision时出现from torchvision import _C ImportError: DLL load failed,其实是在安装torchvision使用命令:pip install --no-deps torchvision安装,导致torchvision与pytoch版本不对应,所以才会出现这样的错误。
文章将记录解决上述错误的解决方法。
下面是我自己安装PyTorch的方法:
需要按照官网上提供的命令安装,官网:https://pytorch.org/get-started/locally/
No CUDA安装方法:
CUD 9.0安装方法:
CUD 8.x安装方法:
CUD 10.0安装方法:
我安装的是CUDA 9.0版本的pytorch,使用上述命令安装时,在pip install torchvision时提示错误,torchvision需要支持的torch>=1.1.0,所以需要再修改pytorch的安装版本,命令如下:
pip3 install https://download.pytorch.org/whl/cu90/torch-1.1.0-cp36-cp36m-win_amd64.whl
按照上述方式安装pytorch后,再使用pip install torchvision安装torchvision就不会报错。
如果你安装的CUDA是9.2版本的话,可以将命令改为:
pip3 install https://download.pytorch.org/whl/cu92/torch-1.1.0-cp36-cp36m-win_amd64.whl
其他版本的CUDA依次类推,注意上述命令为笔者自己猜测的,并没有实际使用安装,读者可以修改cu92和torch-1.1.0的版本号尝试安装。