本文轉載於Ray的pytorch編譯安裝,在此基礎上做了小小的修正。
環境:
MAC OSX 10.13.6 2018 MacBook Pro 15
CUDA 9.2
cuDNN 7.2
顯卡:技嘉 Gaming box 1070
github上別人編譯安裝好的GPU版Pytorch:GPU-Pytorch。別人編譯的和自己機器不是很兼容,所以需要自己來編譯一下。
1. Conda安裝python環境
假設已經安裝好了anconda,這裏我需要的是python3.6,所以直接conda一個環境就可以了。
conda create -n pytorch_gpu_0.4 python=3.6
環境創建好之後,激活一下環境,在這個環境下進行操作。
source activate pytorch_gpu_0.4
2. 下載pytorch源碼
需要下載自己需要版本的pytorch源碼,這裏我需要的是0.4版的,所以把0.4版的clone下來。
git clone --branch v0.4.1 https://github.com/pytorch/pytorch.git
然後進入到文件夾裏
cd <你自己的目錄>/pytorch
最開始clone的時候,submodule的內容並不會自動下載下來,如果不下載下來子模塊的文件的話,會缺少相應的文件。
此時,只需執行如下命令:
git submodule update --init
然後稍微等一會就下完了。
3. 檢查Clang版本
我的Clang版本是9.0,當前的CUDA與Clang版本中的CommandLineTools兼容。如果不兼容,需要去安裝XCODE 9的CommandLineTools。通過這條命令查看clang版本:
$ clang --version
Apple LLVM version 9.0.0 (clang-900.0.39.2)
4. 編譯Pytorch
然後開始編譯GPU版的pytorch,這裏我假設你的CUDA之類的已經都安裝好了。
還有一個需要注意的是,需要安裝一下依賴包:
conda install numpy pyyaml mkl mkl-include setuptools cmake cffi typing
然後直接運行下面這些命令,
export CMAKE_PREFIX_PATH=<換成你的anconda根目錄 >,下面的命令是我的路徑
export CMAKE_PREFIX_PATH=/Users/iimmortall/ProgramFiles/anaconda3
MACOSX_DEPLOYMENT_TARGET=10.9 CC=clang CXX=clang++ python setup.py install
大約20分鐘左右就編譯完成了。
使用源碼安裝 pytorch的時候,是沒有安裝 torchvision的。所以也需要安裝torchvision
pip install torchvision
5.測試Pytorch
我也不知道爲什麼會成爲0.5的版本。。。。。。估計是下載源碼的時候出錯了。
6. 打包生成whl
cd 到pytorch項目中
python setup.py bdist_wheel
生成的whl文件在
/pytorch項目/dist/
我的機器上生成的是 torch-0.5.0a0+a24163a-cp36-cp36m-macosx_10_7_x86_64.whl
Reference
1、github pytorch的官方安裝教程pytorch的官方安裝教程
2、大佬的編譯教程Ray的pytorch編譯安裝