1.安裝說明
本教程使用Anaconda建立Pytorch虛擬環(huán)境來安裝Pytorch教寂。
為什么要使用Anaconda虛擬環(huán)境安裝Pytorch?
- 因?yàn)榄h(huán)境中通常需要安裝很多軟件舔哪,例如:我同時(shí)在使用tensorflow框架臀脏。但是他們所需要的Python的關(guān)聯(lián)模塊或版本會(huì)有所差異虏冻。如果都裝在一個(gè)環(huán)境中難免會(huì)引起沖突肤粱。所以,選擇虛擬環(huán)境能很好地避免環(huán)境之間的沖突厨相。
2.安裝步驟
(1)打開Anaconda Prompt
打開Anaconda Prompt
(2)建立并切換到工作目錄
md \pythonwork
cd \pythonwork
- 后續(xù)Jupyter Notebook讀取與存盤都會(huì)在此工作目錄领曼。
(3)建立Pytorch Anaconda虛擬環(huán)境
conda create --name pytorch python=3.6 anaconda
- 其中加入最后一項(xiàng)anaconda命令建立虛擬環(huán)境時(shí),會(huì)同時(shí)安裝例如Notebook蛮穿、Numpy庶骄、Scipy、Matplotlib践磅、Pandas等數(shù)據(jù)分析包单刁,用戶也可以不加,選擇后續(xù)安裝府适。
(4)啟動(dòng)Pytorch Anaconda虛擬環(huán)境
activate pytorch
(5)安裝Pytorch
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
- 詳情參見Pytorch官網(wǎng)羔飞。
-
具體命令根據(jù)你的環(huán)境選擇。(我安裝的CUDA9.0與Python3.6)
Pytorch官網(wǎng) - 也可以采用其他方法安裝:Pip檐春、LibTorch逻淌、Source。(見上圖自行選擇)
(6)測(cè)試
- 命令行輸入
python
進(jìn)入python疟暖,并輸入下面代碼:
import torch
import torchvision
print(torch.__version__)
- 成功打印卡儒,測(cè)試成功,使用
exit()
退出python俐巴。
(7)關(guān)閉Pytorch Anaconda虛擬環(huán)境
conda deactivate
3.備注
- 到此就安裝完成朋贬,同學(xué)們現(xiàn)在可以愉快地進(jìn)行機(jī)器學(xué)習(xí)項(xiàng)目學(xué)習(xí)與開發(fā)了。(github上有無數(shù)開源代碼窜骄,可以對(duì)感興趣的項(xiàng)目直接進(jìn)行搜索,然后對(duì)項(xiàng)目clone(需安裝git)或直接download摆屯,也可以fork到自己的倉庫(然后使用git pull到本地)邻遏,當(dāng)自己腦子短路或者什么的上去找找靈感吧)
- 在跑別人的項(xiàng)目時(shí)如果遇到相應(yīng)module缺失的情況糠亩,打開Pytorch Anaconda虛擬環(huán)境用conda或pip安裝即可解決。(建議優(yōu)先使用conda准验,conda會(huì)分析依賴包赎线,會(huì)將依賴包一同安裝)
- 如果需要使用本虛擬環(huán)境在Notebook中跑項(xiàng)目,進(jìn)入工作目錄激活虛擬環(huán)境糊饱,輸入Jupyter Notebook運(yùn)行即可垂寥。
- 如果需要使用本虛擬環(huán)境在Pycharm進(jìn)行項(xiàng)目開發(fā),將設(shè)置里的Project Interpreter改為相應(yīng)Anaconda文件目錄下的Pytorch虛擬環(huán)境中的python.exe文件即可另锋。(如:D:\Anaconda3\envs\pytorch\python.exe)
- 本教程為本人學(xué)習(xí)總結(jié)滞项,希望對(duì)同樣在入門學(xué)習(xí)中的你有所幫助。