一、前言
安裝pytorch前,應按照我前邊的博文中,安裝完anaconda和pycharm,并且配置完環境變量以后哈。
Pytorch是什么?
它是一個庫,是一個開源的機器學習框架,專注于深度學習任務,由Facebook的人工智能研究團隊i開發和維護,提供了靈活的張量計算和動態計算圖能力,使得開發者能夠以一種直觀且高效的方式構建和訓練深度學習模型。
優點很多,自行尋找。
然后呢,因為這個庫一直在不斷迭代更新,以及不同的電腦的配置不同,所有就產生了一個適配上的問題,比如說一個配置很低的電腦,肯定是不能去用最新的這個庫,我們應該對不同的電腦選擇不同的庫。
而和深度學習嘻嘻相關的幾個電腦硬件主要是這幾個,GPU(圖形處理單元)、CPU(中央處理單元)、存儲設備(SSD)和內存(RAM)。
而本文主要是講Torch這個庫,而這個庫的安裝主要是和電腦的GPU和CPU這兩個硬件息息相關。所以本文,主要是從這兩個硬件上闡述的。
所以,先檢查一下,自己的電腦有沒有GPU(顯卡)。查詢方式應該很簡單。
二、電腦只有CPU
只有cpu的話就比較簡單了,不會那么復雜了
創建虛擬環境
每一個項目所需要的包都是不同的,就算包相同,那版本也可能會不同,為了避免沖突,所有我們可以為每一個項目配置一個虛擬環境。
1、鍵盤輸入win+R,彈出一個窗口,輸入cmd,點回車進去命令行窗口(如圖所示)
2、創建一個虛擬環境
conda create -n yidaqi python==3.8.5
yidaqi是舉例的一個虛擬環境的名字,python==3.8.5,是選擇的這個虛擬環境中的python版本。
輸入完后,點回車,等待一會,輸入y就行,就會安裝一些基本的庫到所創建的虛擬環境里面。
?3、激活創建的虛擬環境,也就是進入到虛擬環境里面
繼續在剛才創建完的命令行窗口中,輸入以下指令
conda activate yidaqi
就進入到名為yidaqi的虛擬環境當中。
?4、安裝Pytorch
講我們的torch安裝到上邊所創建的虛擬環境里面。
點擊PyTorch
進入pytorch官網,會出現以下界面。
?推薦使用conda命令,conda是anaconda的包管理器。
(1)conda可以讓你同時管理安裝處理有關的python任務和跟python無關任務,即pip可以允許在任何環境中安裝python包,conda允許你在conda環境中安裝任何語言包(包括C語言或者python)。
(2)因為pip不能安裝和解析conda的包格式。可以使用這兩個工具,但是它們是不能交互的?
(3)由于pip是從源代碼安裝的,所以如果你無法編譯源代碼,那么安裝它可能會很痛苦(這在Windows上尤其如此,但如果軟件包有一些困難的C或FORTRAN庫,甚至在Linux上也是如此)依賴)。?
(4)Conda從二進制文件安裝,這意味著有人(例如Continuum)已經完成了編譯軟件包的艱苦工作,因此安裝非常簡單。不過,個人感覺conda安裝有些慢。(相對pip 來說)
安裝pytorch
出現
5、安裝完成后,在創建的虛擬環境下,輸入conda list看一下有哪些庫。
這三個出來就沒問題了。使用conda安裝出現的是pytorch、torchvision和torchaudio。
使用pip安裝出現的應該是torch、torchvision和torchaudio。
也可以用下面方法測試是否安裝成功,測試方法為:
? 測試numpy: import numpy
? ? 測試pytorch: import torch
若輸入import numpy 報錯:需下載numpy包,安裝完之后再測試就可以了。
可輸入命令:conda install numpy,