您好,登錄后才能下訂單哦!
這篇文章給大家分享的是有關(guān)PyTorch如何安裝配置的內(nèi)容。小編覺得挺實(shí)用的,因此分享給大家做個(gè)參考,一起跟隨小編過來看看吧。
1. 環(huán)境說明
PyTorch目前支持OSX和Linux兩種系統(tǒng),并且支持多種安裝方式。在官網(wǎng)上介紹了基于conda,pip和源代碼編譯幾種不同的安裝方式。支持的Python版本有2.7,3.5和3.6。鑒于深度學(xué)習(xí)需要的計(jì)算量一般比較大,強(qiáng)烈建議找到一個(gè)有獨(dú)立顯卡的電腦來展開學(xué)習(xí),當(dāng)然沒有顯卡也能用,就是計(jì)算慢很多了。
如果你的電腦是Windows,那么可以安裝一個(gè)虛擬機(jī)來運(yùn)行Linux,但是性能可能會(huì)折扣。我的電腦操作系統(tǒng)是Ubuntu16.04,所以我會(huì)以此為例子來介紹后面的內(nèi)容。
2. Anaconda 和Python
PackageManager我們選擇conda,于是我們需要安裝Anaconda這個(gè)功能強(qiáng)大的包,下載地質(zhì):https://www.anaconda.com/download/ 。里面包含了conda工具,也有Python,以及很多Python需要的擴(kuò)展工具包。選擇2.7版本,下載并安裝即可。接下來,如果有顯卡,進(jìn)入第3步;沒有顯卡直接到第4步。
3. 顯卡驅(qū)動(dòng)和CUDA
要使用顯卡進(jìn)行運(yùn)算,你需要使用支持CUDA的NVIDIA顯卡,目前比較好的顯卡有NVIDIATITANX、GTX1080Ti 等。好的顯卡將會(huì)是深度學(xué)習(xí)研究的有力武器。當(dāng)然,普通的顯卡例如GTX970、GTX1060等也是可以用的。實(shí)在沒有顯卡,那只能做簡(jiǎn)單的小數(shù)據(jù)量的實(shí)驗(yàn),效果不會(huì)太好。
在Ubuntu16.04上可以采用以下方法安裝顯卡驅(qū)動(dòng),這種方式比較穩(wěn)定。打開“SystemSettings”——“Software&Updates” —— “AdditionalDrivers”,聯(lián)網(wǎng)狀態(tài)會(huì)自動(dòng)搜索可用的顯卡驅(qū)動(dòng),選擇可用的版本,點(diǎn)擊“ApplyChanges”即可??赡苤貑?dòng)后顯卡驅(qū)動(dòng)才能生效。
接下來安裝CUDAToolkit。因?yàn)樽钚碌腃UDA是9.0版本,但是PyTorch只能支持到8.0。所以轉(zhuǎn)到該網(wǎng)址“https://developer.nvidia.com/cuda-80-ga2-download-archive”,選擇對(duì)應(yīng)的參數(shù),下方會(huì)有對(duì)應(yīng)的CUDA文件。例如,下圖是我選擇的版本。選擇deb文件,發(fā)現(xiàn)有1.9G大小,下載下來,按照baseinstaller的指示來完成安裝。附加的cuBLAS也可以下載下來安裝上。
CUDA安裝完成后,在主目錄下打開“.bachrc”文件,在末尾添加如下代碼:
export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
4. 運(yùn)行PyTorch的安裝命令
如果你和我一樣,采用8.0CUDA,2.7版本Python,可以運(yùn)行以下命令:
conda install pytorch torchvision cuda80 -c soumith
如果沒有顯卡,2.7版本Python,可以這樣:
conda install pytorch torchvision -c soumith
5. 驗(yàn)證安裝是否成功
要顯示顯卡信息,在終端輸入:
sudo lshw -c video
要顯示CUDA信息,在終端輸入:
nvcc -V
查看Phthon版本,在終端輸入:
python --version
驗(yàn)證pytorch是否安裝成功,在終端輸入:
python
此時(shí)進(jìn)入python環(huán)境。然后,繼續(xù)輸入
import torch import torchvision
不報(bào)錯(cuò)就表明安裝成功。
感謝各位的閱讀!關(guān)于“PyTorch如何安裝配置”這篇文章就分享到這里了,希望以上內(nèi)容可以對(duì)大家有一定的幫助,讓大家可以學(xué)到更多知識(shí),如果覺得文章不錯(cuò),可以把它分享出去讓更多的人看到吧!
免責(zé)聲明:本站發(fā)布的內(nèi)容(圖片、視頻和文字)以原創(chuàng)、轉(zhuǎn)載和分享為主,文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng),如果涉及侵權(quán)請(qǐng)聯(lián)系站長(zhǎng)郵箱:is@yisu.com進(jìn)行舉報(bào),并提供相關(guān)證據(jù),一經(jīng)查實(shí),將立刻刪除涉嫌侵權(quán)內(nèi)容。