MacOS 10.13.3 Cuda 9.1 Cudnn 7.0.5 Pytorch 從零編譯

-- 6.26 更新

真是不作死就不會死,之前用brew 安裝了open-mpi 庫,但是更新brew 庫之後,Open-mpi 庫沒裝上,編譯Pytorch 就總出現問題。

換了8700K 之後 全核心5GHZ 的速度,編譯起C++代碼來速度真的飛起。期待9700K 的到來。


-- 6.14 更新

之前編譯的是0.4 的內部版本,很多函數還是和正式版有所差別的

由於Pytorch 更新到0.4 版本,很多函數相對變化。對於黑蘋果用戶或者egpu 用戶來說,我們需要重新編譯。編譯的步驟也有所不同。現在已經改爲如下。

由於MacOS 10.13.4 更新的幅度很大,支持第三方Nnme SSD ,並且很多方面做了優化。所以對於黑蘋果用戶來說升級的作用很大。但是由於當時手賤升級之後也沒備份,也沒保存,所以我們當時一氣之下將系統刪除,而現在,cuda 更新到9.1 的版本,發現各大深度學習庫都對新Cuda 進行了支持,所以又將黑果裝上了,目前裝黑果也是傻瓜式安裝了,只要按照步驟來,用指定的硬件,基本上不會出什麼大問題。需要注意的地方我已經在另一個博客中予以了說明。

現在cuda 9.1  和xcode 9.2 也是搭配的很好,不需要單獨去選了


至於TensorFlow 的安裝,GitHub 上有很多教程,也有編譯好的whl 文件可以直接進行安裝。而Pytorch 則沒有,所以需要源碼編譯,而源碼編譯的速度也很快,感覺比TensorFlow 要輕量級很多。

至於要注意的地方就是

conda install numpy pyyaml mkl mkl-include setuptools cmake cffi typing

同時執行下面的命令即可

export CMAKE_PREFIX_PATH='/Users/zhao/anaconda3'

export CUDNN_INCLUDE_DIR="/Developer/cuda/include/"

export CUDNN_LIB_DIR="/Developer/cuda/lib/"

MACOSX_DEPLOYMENT_TARGET=10.9 CC=clang CXX=clang++ python setup.py install

最後一句必須這樣寫,如果不寫前面的則會報錯。


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