Windows安装PyTorch方法

    在使用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的版本号尝试安装。