PyTorch安裝流水賬 win10按默認步驟安裝Anaconda後各指令狀況&Anaconda配置環境變量

背景

win10系統,15年的老電腦,GPU是NVIDIA GeForce GTX 960M。

安裝前做了一些調研,PyTorch分爲GPU版本和CPU版本,設想是安裝GPU版本,然後可以在本地跑一些小demo,之後租雲服務器跑程序。

Win10查看NVIDIA顯卡GPU利用率和溫度,順便還可以看CUDA版本

參考:https://www.cvmart.net/community/detail/3493

先弄懂一個GPU相關的概念——CUDA:簡單來說,這是英偉達開發的一個編程接口層,能讓你調用GPU的指令集及其並行計算單元。[用 Windows 電腦訓練深度學習模型?]

打開cmd輸入下面指令:

cd C:\Program Files\NVIDIA Corporation\NVSMI

接着再執行nvidia-smi.exe命令,即可查看顯卡的整體信息,下面是我電腦的情況:

然後我來到pytorch的官網,發現最新支持的CUDA已經是11.6而我的是11.4考慮到各種麻煩,改裝CPU版本。

安裝工具以及虛擬環境的創建

根據官網,它是有兩種方式安裝,一個是pip和一個是Anaconda,Anaconda我之前已經裝了,並且深深體會到它的好用。

見[win10按默認步驟安裝Anaconda後各指令狀況&Anaconda配置環境變量]

目前我電腦上的Anaconda版本是3 5.2.0,默認Python版本則是3.9.7

conda是一個包和環境管理器,安裝anaconda的時候就一起包含在裏面了,用conda創建一個虛擬環境:

conda create --name env4pytorch

如果這一步頻繁報錯則必須使用鏡像大法!見[https://www.cnblogs.com/xkxf/p/15952815.html]

弄好之後我們active這個環境:

conda activate env4pytorch

這纔開始安裝PyTorch

直接拷貝官網的指令:

conda install pytorch torchvision torchaudio cpuonly -c pytorch

等待好一段時間。終於安裝好,測試一下。輸入python進入一個Python的交互環境,輸入下面代碼:

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

輸出

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]])

大功告成!

小結

Pytorch有兩個版本,一個是cpu版本一個gpu版本,

gpu版本一般要有英偉達顯卡,並且光有GPU不行,GPU對應的軟件CUDA版本還得跟上!

有了軟硬件準備,安裝Pytorch有兩個方式,個人感覺更好的是Anaconda全家桶,

創建虛擬環境,激活環境,這樣不容易把環境搞亂,大不了把環境刪了。

最後鍵入官方網站的對應指令即可!

但是其實過程中的,鏡像大法啥的弄來弄去也得耗些時間!

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章