Linux下便捷安裝 Pytorch命令指南

Pytorch是Facebook的開源深度學習框架,目前還不支援Windows系統,僅支援Linux和Mac。不過不要緊,深度學習的從業者多數習慣在Linux環境下進行相關的研究、開發。Pytorch的安裝還是很簡單的,官網的安裝指導簡單明瞭:點選選擇按鈕,生成安裝命令。但問題是,有時候需要翻牆。這種安裝方式的另外一個缺點是,原始碼被封裝成
.so檔案,看不到內部實現細節。因此,最好的安裝方式其實是從GitHub下載原始碼,然後在一步步安裝。這種安裝方式的問題是需要事先安裝各種依賴庫,安裝過程很不“傻瓜”,因此有了本博文,寫給那些既不想手動從原始碼安裝,又懶得爬牆的童鞋。

方式一、Python2.7,無CUDA

1.pip installhttp://download.pytorch.org/whl/cu75/torch-0.1.12.post2-cp27-none-linux_x86_64.whl 

2.pip install torchvision

方式二、Python 2.7,CUDA 8.0

1.pip install http://download.pytorch.org/whl/cu80/torch-0.1.12.post2-cp27-none-linux_x86_64.whl 
2.pip install torchvision

方式三、Python 3.5, 無CUDA

1.pip
install http://download.pytorch.org/whl/cu75/torch-0.1.12.post2-cp35-cp35m-linux_x86_64.whl 
2.pip install torchvision

方式四、Python
3.5,CUDA 8.0

1.pip
install http://download.pytorch.org/whl/cu80/torch-0.1.12.post2-cp35-cp35m-linux_x86_64.whl 
2.pip install torchvision