1. 程式人生 > >在Windows上安裝pytorch

在Windows上安裝pytorch

clas dll color rar 命令 tar license uda 文件夾

電腦環境為 :windows10 python3.5 cuda8.0

可以在官網查詢各個版本的安裝方式

技術分享圖片

這裏使用pip的安裝方式,通過以下命令:

1 pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.0-cp35-cp35m-win_amd64.whl 
2 pip3 install torchvision

在安裝完成後在python中驗證安裝:

Python 3.5.2 (v3.5.2:4def2a2901a5, Jun 25 2016, 22:18:55) [MSC v.1900 64 bit (AMD64)] on win32
Type 
"help", "copyright", "credits" or "license" for more information. >>> import torch >>> import torchvision >>>

可能出現的問題:

在import torch時出現 from torch._C import * ImportError: DLL load failed: 找不到指定的模塊 的錯誤

原因是缺少一些dll文件,通過以下幾步來解決:

1.下載 win-64/intel-openmp-2018.0.0-8.tar.bz2

技術分享圖片

2.將其解壓,並將其Library/bin文件夾中的dll文件全部放入python所在目錄的DLLs文件夾下(例如C:\Users\qin\AppData\Local\Programs\Python\Python35\DLLs)

3.將上面的路徑添加進環境變量path中。

OK,完成

在Windows上安裝pytorch