這篇文章給大家分享的是有關PyTorch如何安裝配置的內(nèi)容。小編覺得挺實用的，因此分享給大家做個參考，一起跟隨小編過來看看吧。
創(chuàng)新互聯(lián)公司服務項目包括龍城網(wǎng)站建設、龍城網(wǎng)站制作、龍城網(wǎng)頁制作以及龍城網(wǎng)絡營銷策劃等。多年來，我們專注于互聯(lián)網(wǎng)行業(yè)，利用自身積累的技術優(yōu)勢、行業(yè)經(jīng)驗、深度合作伙伴關系等，向廣大中小型企業(yè)、政府機構等提供互聯(lián)網(wǎng)行業(yè)的解決方案，龍城網(wǎng)站推廣取得了明顯的社會效益與經(jīng)濟效益。目前，我們服務的客戶以成都為中心已經(jīng)輻射到龍城省份的部分城市，未來相信會繼續(xù)擴大服務區(qū)域并繼續(xù)獲得客戶的支持與信任！1. 環(huán)境說明
PyTorch目前支持OSX和Linux兩種系統(tǒng)，并且支持多種安裝方式。在官網(wǎng)上介紹了基于conda，pip和源代碼編譯幾種不同的安裝方式。支持的Python版本有2.7，3.5和3.6。鑒于深度學習需要的計算量一般比較大，強烈建議找到一個有獨立顯卡的電腦來展開學習，當然沒有顯卡也能用，就是計算慢很多了。
如果你的電腦是Windows，那么可以安裝一個虛擬機來運行Linux，但是性能可能會折扣。我的電腦操作系統(tǒng)是Ubuntu16.04，所以我會以此為例子來介紹后面的內(nèi)容。
2. Anaconda 和Python
PackageManager我們選擇conda，于是我們需要安裝Anaconda這個功能強大的包，下載地質(zhì)：https://www.anaconda.com/download/ 。里面包含了conda工具，也有Python，以及很多Python需要的擴展工具包。選擇2.7版本，下載并安裝即可。接下來，如果有顯卡，進入第3步；沒有顯卡直接到第4步。
3. 顯卡驅(qū)動和CUDA
要使用顯卡進行運算，你需要使用支持CUDA的NVIDIA顯卡，目前比較好的顯卡有NVIDIATITANX、GTX1080Ti 等。好的顯卡將會是深度學習研究的有力武器。當然，普通的顯卡例如GTX970、GTX1060等也是可以用的。實在沒有顯卡，那只能做簡單的小數(shù)據(jù)量的實驗，效果不會太好。
在Ubuntu16.04上可以采用以下方法安裝顯卡驅(qū)動，這種方式比較穩(wěn)定。打開“SystemSettings”——“Software&Updates” —— “AdditionalDrivers”，聯(lián)網(wǎng)狀態(tài)會自動搜索可用的顯卡驅(qū)動，選擇可用的版本，點擊“ApplyChanges”即可?？赡苤貑雍箫@卡驅(qū)動才能生效。
接下來安裝CUDAToolkit。因為最新的CUDA是9.0版本，但是PyTorch只能支持到8.0。所以轉(zhuǎn)到該網(wǎng)址“https://developer.nvidia.com/cuda-80-ga2-download-archive”，選擇對應的參數(shù)，下方會有對應的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. 運行PyTorch的安裝命令
如果你和我一樣，采用8.0CUDA，2.7版本Python，可以運行以下命令：
conda install pytorch torchvision cuda80 -c soumith
如果沒有顯卡，2.7版本Python，可以這樣：
conda install pytorch torchvision -c soumith
5. 驗證安裝是否成功
要顯示顯卡信息，在終端輸入：
sudo lshw -c video
要顯示CUDA信息，在終端輸入：
nvcc -V
查看Phthon版本，在終端輸入：
python --version
驗證pytorch是否安裝成功，在終端輸入：
python
此時進入python環(huán)境。然后，繼續(xù)輸入
import torch import torchvision
不報錯就表明安裝成功。
感謝各位的閱讀！關于“PyTorch如何安裝配置”這篇文章就分享到這里了，希望以上內(nèi)容可以對大家有一定的幫助，讓大家可以學到更多知識，如果覺得文章不錯，可以把它分享出去讓更多的人看到吧！
另外有需要云服務器可以了解下創(chuàng)新互聯(lián)scvps.cn，海內(nèi)外云服務器15元起步，三天無理由+7*72小時售后在線，公司持有idc許可證，提供“云服務器、裸金屬服務器、高防服務器、香港服務器、美國服務器、虛擬主機、免備案服務器”等云主機租用服務以及企業(yè)上云的綜合解決方案，具有“安全穩(wěn)定、簡單易用、服務可用性高、性價比高”等特點與優(yōu)勢，專為企業(yè)上云打造定制，能夠滿足用戶豐富、多元化的應用場景需求。
