天天看點

linux github安裝 Pytorch

前言

因為網絡原因(你懂得牆),用官網的安裝辦法根本無法安裝pytorch,無奈去github看到了源碼安裝。話不多說,開始安裝吧

通路github

pytorch在github上的位址

pytorch有源碼安裝的教程:

github推薦了2種安裝方法

  • 官網上的3種方法(conda,pip等)
  • 源碼安裝

這裡就介紹Install From Source

準備工作1:安裝anaconda

官網也強力推薦anaconda,這個可以作為python版本管理工具,真心非常好用,而且內建各種第三方庫。

If you are installing from source, we highly recommend installing an Anaconda environment. You will get a high-quality BLAS library (MKL) and you get a controlled compiler version regardless of your Linux distro.

但是還是那個原因,下載下傳太慢了,可以從我的百度雲下載下傳

百度雲連結

密碼:lotq

其中有個要注意的,就是anaconda的配置。就是加入你錯過了點選yes的機會,anaconda就不會幫你配置好環境變量了。是以要進行以下操作:

# 将anaconda的bin目錄加入PATH
echo 'export PATH="~/anaconda3/bin:$PATH"' >> ~/.bashrc
# 更新bashrc以立即生效
source ~/.bashrc           

最後确認一下,如圖

準備工作2:配置一下并且安裝依賴

export CMAKE_PREFIX_PATH="$(dirname $(which conda))/../" # [anaconda root directory]
# 上面這種方法經常不好用,使用下面兩種方法:
echo 'export CMAKE_PREFIX_PATH="~/anaconda3/bin"' >> ~/.bashrc
# 或者粘貼下一行代碼到.bashrc中
export CMAKE_PREFIX_PATH=~/anaconda3/bin
# 然後
source ~/.bashrc           

打開.bashrc檔案,看到如圖所示:

其實就是修改配置檔案,直接在配置檔案加上這句話就好啦。

# 安裝基本的依賴
conda install numpy pyyaml mkl setuptools cmake gcc cffi

# Add LAPACK support for the GPU
conda install -c soumith magma-cuda80 # or magma-cuda75 if CUDA 7.5           

(其實這個也挺慢的)

安裝:clone并安裝

安裝成功後,就把項目clone下來吧

git clone --recursive https://github.com/pytorch/pytorch           

然後

python setup.py install           

終于安裝好了,挺麻煩的,最好還是科學上網,然後用官網的方法安裝吧