MAC OSX 編譯安裝GPU版的Pytorch CUDA9.2 cuDNN7.2

本文轉載於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編譯安裝

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章