windows下pytorch的安裝

PyTorch的安裝

1.下載windows版本的conda包

conda包下載地址：https://pan.baidu.com/s/1dF6ayLr#list/path=%2Fpytorch。大家根據需要選擇相應的檔案下載。

注：檔名的解釋：一般為PACKAGENAME-VERSION-PYTHON_VERSION cu80.tar.bz

PACKAGENAME 分為 pytorch 和 pytorch_legacy， 分別為NT核心版本10和6的兩類系統進行編譯；VERSION代表 pytorch 的版本；而PYTHON則代表python程式的版本，主要分為3.5和3.6。

2.安裝conda包

開啟命令列提示符，cd到conda包下載的位置。然後執行下面兩個命令：

conda install numpy mkl cffi
conda install --offline pytorch-0.2.1-py36he6bf560_0.2.1cu80.tar

•  

3.測試PyTorch

附一段簡單測試CUDA與cuDNN是否工作正常的程式碼：

# CUDA TEST
import torch
x = torch.Tensor([1.0])
xx = x.cuda()
print(xx)

# CUDNN TEST
from torch.backends import cudnn
print(cudnn.is_acceptable(xx))

如果能正常輸出結果則表明安裝成功。

4.import torch出錯處理

如果時原來已經安裝了anaconda3，在import torch時可能產生如下錯誤：

Traceback (most recent call last):
  File "test.py", line 2, in <module>
    import torch
  File "C:\Anaconda3\lib\site-packages\torch\__init__.py", line 41, in <module>
    from torch._C import *
ImportError: DLL load failed: The specified module could not be found.

這時我們只需要更新一下Anaconda3的python版本至3.6.1及以上即可。更新的方式有以下兩種：

conda install -c anaconda python=3.6.1 # 更新至指定版本
conda updata python # 直接更新