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的版本號嘗試安裝。