首頁 > 軟體

安裝PyTorch的詳細過程記錄

2022-03-10 19:23:36

安裝anaconda

登入anaconda的官網下載,anaconda是一個整合的工具軟體不需要我們再次下載。anaconda官網

點選下載

跳轉到這個頁面如果你的Python版本正好是3.8版,那便可以直接根據系統去選擇自己相應的下載版本就可以了。

但是如果你的Python版本號不是當前頁面的版本號,那我建議你去選擇相對應的版本號。

點選archive你就會跳轉到下面的頁面

你可以存取這篇部落格去找到當前與你python版本號相對應的anaconda(比如我的python是3.7.2版本的,因為這個版本在那篇部落格中沒有對應的anaconda的版本所以我便選擇了Anaconda3-2019.03,而這個版本對應的是python3.7.3版,新的版本對舊的版本具有包容性,雖然有的人說運用最新的版本對現在所有的python版本都可以相容,但是我沒有試過,如果有試過的可以告訴我)。

如果不知道怎麼查詢版本號?同時按住win+R進入cmd輸入命令: python --version

下載完便開始安裝,根據你的需求去選擇,看你是商業用途還是個人用途,我選擇的是個人

一定要記住自己安裝的路徑

點選對勾將anaconda的預設值設定為python3.7

剩下的預設就可以了。

安裝完成以後點選win圖片,在最近新增和anaconda包中都能找到anaconda prompt。

出現(base)便表示成功

環境管理

在我們做專案的時候可以能需要不同環境的python版本,有時候要1.0版本,有的需要3.0版本的拿在這裡我們就需要建立不同的環境,在不同的需要的時候去使用。
這個是在Anaconda Prompt下操作的

conda create -n

這是建立的格式

"pytorch"是這個環境變數的名字

"python=3.7"是我們要確定的當前環境的版本數

告訴我們建立這個環境需要這些包加入,同意就輸入y

這樣就操作成功。

我們輸入activate pytorch(pytorch是你定義的這個環境的名字),左邊的環境就從base(基本環境),變成了pytorch環境。

輸入pip list,檢視當前環境下面有哪些包,我們發現沒有pytorch,那麼下面我們就需要安裝它。

PyTorch安裝

進入pytorch官網

複製這一段操作指令。(這種情況是需要電腦上有單獨的英偉達的顯示卡、或者英偉達的顯示卡和集顯這兩種情況都是可以的。)

進入命令符號視窗,輸入nvidia-smi,檢視當前驅動的版本號,觀察Driver Version的值是否大於400,如果小於請更新顯示卡驅動。推薦使用驅動精靈更新,雖然驅動精靈的版本沒有官網的更新的那麼及時但是驅動精靈更新簡單,不需要過多的操作。當然也可以去英偉達顯示卡驅動更新去下載。

選擇自己顯示卡相對應的系列,括號中Notebooks是筆電。

將上邊複製的程式碼貼上進去。注意此事的環境是pytorch

輸入y

便開始下載了(下載時間較長請耐心等待)

下載完成後我們再次輸入pip list,檢視到已經有torch的存在。

檢驗安裝

先輸入python,然後輸入import torch,如果輸入後沒有任何報錯,沒有任何顯示那就是成功了,然後再輸入torch.cuda.is_available(),返回的是True,那便是完成了整個操作。

上述步驟參考https://www.bilibili.com/video/BV1hE411t7RN?t=734,中間參雜自己操作時的一些操作和思考。

總結

到此這篇關於安裝PyTorch的文章就介紹到這了,更多相關安裝PyTorch過程內容請搜尋it145.com以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援it145.com!


IT145.com E-mail:sddin#qq.com