1. 程式人生 > 實用技巧 >pytorch環境的配置及安裝

pytorch環境的配置及安裝

1. Anaconda 下載

Anaconda是一個集成了常用於科學分析(機器學習,深度學習)的大量package(安裝包)。也就是說,你只要安裝了 Anaconda,就安裝了很多我們之後要用的許多packages。

Anacode 下載地址:https://www.anaconda.com/products/individual

2. Anaconda 安裝

雙擊進行安裝,需要注意以下幾點:

(1)記住安裝路徑(之後會用到)

(2)跳過安裝 Microsoft VSCode

為了檢驗是否安裝成功,在開始選單出,左擊 Anaconda Prompt

如果可以成功開啟,且左邊有 (base),即安裝成功

3. 顯示卡配置(無 Nvidia 顯示卡的略過)

只要你開啟工作管理員,在 GPU 那裡看到了你的 NVIDIA 顯示卡,即可。說明你的硬體驅動,已安裝。

如果你的 NVIDIA 顯示卡未顯示,可以使用驅動人生,或者去官網下載驅動。

4. 有序的管理環境

你之後會遇到不同的專案,需要使用到不同版本的環境。比如這個專案要用到 pytorch 0.4,另一個專案要用到 pytorch 1.0,如果你解除安裝了0.4版本,安裝了1.0版本。那麼下一次,你再碰到0.4版本,你就需要解除安裝1.0版本,安裝0.4版本。很折騰。

Anaconda 整合的 conda 包就能夠解決這個問題。它可以創造出兩個屋子,相互隔離。一個屋子放 0.4 版本,一個屋子放 1.0 版本。你需要哪個版本,就進哪個屋子工作。

我們首先使用 conda 指令建立一個屋子,叫做 pytorch。

指令如下:

conda create -n pytorch python=3.6

conda 是指呼叫 conda 包,create 是建立的意思,-n 是指後面的名字是屋子的名字, pytorch是屋子的名字(可以更改成自己喜歡的),python=3.6 是指建立的屋子,是 python3.6 版本。

之後,彈出提示,輸入 y,即可安裝。

下面兩個命令是進入和退出某個環境的命令:

安裝成功後,輸入以下指令:

conda info --envs

即可看到 conda 環境中,有新建的 pytorch 環境,右邊的 * 號表示,當前你處於哪個環境。

接下來,我們要在 pytorch 環境中,安裝 PyTorch,(有點繞),使用如下指令,進入 pytorch 環境。

conda activate pytorch

你可以看到左邊的 base 變成了 pytorch,代表成功進入 pytorch 環境。

5. 安裝 PyTorch 準備

PyTorch官網地址:https://pytorch.org/

PyTorch Build 選擇 Stable;

選擇系統;

Package,Windows下推薦 conda,Linux 下推薦 pip;

Python版本按照Anaconda的版本選擇,我這裡選擇3.6;

CUDA 推薦9.2。如果沒有顯示卡的話,選擇 None。

複製下面的程式碼,之後,在開始選單中,開啟 Anaconda Prompt,檢視最左邊括號中是 base 還是 pytorch。

(如果是 base,使用conda activate pytorch進入 pytorch 環境中)

果斷輸入y,之後就是漫長的等待。或者去休息吧,慢慢等著。

6. 驗證是否安裝成功

(1)在命令列左邊為 pytorch 環境中,輸入 python

(2)之後,輸入import torch,如果沒有報錯,意味著 PyTorch 已經順利安裝了

(3)接下來,輸入torch.cuda.is_available(),如果是 True,意味著你可以使用 GPU,如果是 False,意味著只能使用CPU。