Pytorch在Windows下的安裝

目錄

• 前言
• 目前Pytorch支援的windows系統
• 可供選擇的Pytorch安裝方式
• 安裝過程及注意事項
• 測試是否安裝成功

前言

最近Pytorch很火，相比tensorflow，更加簡單明了，是以躍躍欲試地在本就不寬裕的硬碟空間下安裝完成，整體過程很簡單，但還是有一些要注意的地方需要總結。

目前Pytorch支援的windows系統

最新穩定版的Pytorch1.0版本已經支援三種類型的作業系統，Linux、Mac以及Windows，這樣大大增加了大家的選擇餘地。

不過針對Windows系統，目前隻支援Win7以上作業系統，當然這是最低要求，官方文檔的描述如下：

Windows 7

and greater;

Windows 10

or greater recommended.

Windows Server 2008 r2

and greater

So,如果大家系統還沒達到要求的話，第一步是要更新你的系統。

可供選擇的Pytorch安裝方式

可供選擇的安裝方式大體有三種：

1. Anaconda

2. pip

3. BUILDING FROM SOURCE

我本人分别在Ubuntu和Windows系統下安裝過Tensorflow，從我自身的使用感受來看，Anaconda确實是一個相對簡單友善的方式，也更加友善後期的管理，是以這次還是選擇了這種方式。

建議大家如果不想過多踩坑的話，選擇這種方式沒錯啦。

安裝過程及注意事項

前期當然是要安裝系統必備的應用，anaconda、python之類的，這裡就不贅述了。

打開anaconda的prompt，然後安裝指令一般就是一句話：

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

當然要結合你的系統環境。

系統一般會自動下載下傳必備的元件并安裝，但是嘗試了幾次都會報HTTPError。

因為現在連接配接的是預設源，即使連接配接成功也隻是每秒十幾k的下載下傳速度。

是以我們需要更改預設的連接配接源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

           

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
           

連接配接清華源以後，下載下傳速度會大幅度提升，一般很快會順利安裝完成。

但我昨天安裝的過程中，一直出現一個元件無法下載下傳完成

是以這裡提示大家，可以手動下載下傳這個元件，并選擇本地安裝，同樣可以順利完成整個Pytorch的配置，使用指令如下。

conda install --use-local your-pkg-name
           

測試是否安裝成功

在你的Python編輯器編輯代碼：

from __future__ import print_function
import torch
x = torch.rand(5, 3)
print(x)
           

如果輸出與下面類似，就說明安裝成功了，這是輸出了一個5*3的随機tensor。

tensor([[0.3380, 0.3845, 0.3217],
        [0.8337, 0.9050, 0.2650],
        [0.2979, 0.7141, 0.9069],
        [0.1449, 0.1132, 0.1375],
        [0.4675, 0.3947, 0.1426]])
           

如果還想測試你的GPU和CUDA工作良好的話，可以輸入如下代碼：

import torch
torch.cuda.is_available()
           

如果輸出是True的話，就代表你的系統環境完全沒問題啦，enjoy你的深度之旅吧！