1. 程式人生 > >gcc4.8+Caffe2(0.8)+onnx(1.2)+onnx-caffe2在anaconda中的安裝,不需要root,原始碼編譯.

gcc4.8+Caffe2(0.8)+onnx(1.2)+onnx-caffe2在anaconda中的安裝,不需要root,原始碼編譯.

caffe2的cuda8預編譯是gcc5以上的. gcc5以下的只有cuda9和cpu版本

對於gcc4.8和cuda8的caffe2沒有預編譯. 

原始碼安裝caffe2

git clone --recursive https://github.com/pytorch/pytorch.git
cd pytorch
git submodule update --init
CONDA_INSTALL_LOCALLY=1 ./scripts/build_anaconda.sh --cuda 8.0 --cudnn 7 -DUSE_CUDA=ON -DUSE_NCCL=ON

由於onnx-caffe2已經整合在caffe2中,所以不用再安裝

原始碼安裝onnx

git clone https://github.com/onnx/onnx.git --recurrence
pip install ./onnx