前言
因爲網絡原因(你懂得牆),用官網的安裝辦法根本無法安裝pytorch,無奈去github看到了源碼安裝。話不多說,開始安裝吧
訪問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
終於安裝好了,挺麻煩的,最好還是科學上網,然後用官網的方法安裝吧