PyTorch環境配置及安裝

環境配置


舒適提示:爲了更好的教程體驗,提供視頻、閱讀地址python

Youtube: https://www.youtube.com/playlist?list=PLgAyVnrNJ96CqYdjZ8v9YjQvCBcK5PZ-Vgit

BiliBili: https://www.bilibili.com/video/av74281036/github

Github: https://github.com/xiaotudui/PyTorch-Tutorial機器學習

相關下載:https://pan.baidu.com/s/16koDDDvAFO7rNoi-uR-A_g 提取碼:pahiide

在學習中,遇到任何問題,能夠關注公衆號:土堆碎念,直接在對話框中留言,我會不定時解答的~函數

公衆號:土堆碎念,回覆:pytorch,便可得到全部的數據集及安裝包, 省去找軟件、數據集的時間。(同步更新中,請勿急躁)工具


0. 前言

在計算機領域學習,最煩的一件事,就是安裝軟件,環境配置。並且,偏偏不如意的是,它仍是你學習進程的第一關。第一關就出現大Boss。這樣,超級容易打擊學習的動力。學習

因此,在我看來,入門的話,必定要快,要不求甚解,先把握總體。因此,個人教程安排順序一些方法,不一樣於其餘千篇一概的教程,其中也許有本身的想法。ui

若是你在學習中,有什麼建議,或者遇到什麼問題,能夠關注公衆號:土堆碎念,直接留言,私信我。spa

1. Anaconda 下載

在機器學習,深度學習中,要用到大量的 package(就是各類工具包)。若是說,函數是一個工具,那麼 package 就是一個工具包。一個個安裝 package 很麻煩,並且容易出現疏漏。因而,就有了 Anaconda,這是一個集成了經常使用於科學分析(機器學習,深度學習)的大量package。

也就是說,你只要安裝了 Anaconda,就安裝了不少咱們以後要用的許多packages。(他還有不少功能,對入門幫助超大,後面再說)

Anaconda 的下載地址:這兒(目前,頁面上是 Python3.7 的版本,我用的是3.6版本,推薦3.6版本

以前使用過3.7版本,也許會遇到一些問題。

推薦: Anaconda歷史版本連接:這兒,選擇這個版本: Anaconda 3-5.2.0

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了。激動的打開官網,下拉到下面的頁面。

(不得不說,PyTorch 在這方面作的真的好,不須要再人工安裝 CUDA、cuDNN 之類的,所有都給你解決了。真的良心~之前安裝真的繁瑣。(吐槽一句,之前超麻煩的)

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. 加速(可選)

有的時候,下載速度太慢了,能夠利用我已經下載好的文件。

連接:https://pan.baidu.com/s/1cyEjHdluc4ufCkvtbgziqA 
提取碼:bzfx 
複製這段內容後打開百度網盤手機App,操做更方便哦

將其中的 pytorch-1.3.0-py3.6_cuda92_cudnn7_0.tar 和 cudatoolkit-9.2-0.tar兩個文件,複製到剛纔 Anaconda 安裝的目錄下的 pkgs 文件夾中。

這兩個文件,只能安裝 pytorch1.3.0 和 cuda 9.2,僅限Windows。

以後,在 Anaconda Prompt 命令行窗口,多按幾回 Ctrl + C 結束程序。而後,從新複製 PyTorch 安裝命令,粘貼到命令行上,進行安裝。

7. 驗證是否安裝成功

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

(2)以後,輸入 import torch,若是沒有報錯,意味着 PyTorch 已經順利安裝了

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


往期推薦

學會這三個排版原則,你就能超過80%人的排版

你想過將你的人生遊戲化嗎?

用遊戲的思路激勵本身

尋找有趣或更有效率的事、工具和教程

相關文章
相關標籤/搜索