Skip to content

Install

Hiroki Naganuma edited this page Jul 12, 2017 · 2 revisions
cd nv_caffe_home
rm -rf build
mkdir build
cd build
LOCAL=/home/hiroki11/env/local

declare -a PACKAGES=(\
'atlas_3.10.3' \
'boost_1.63.0' \
'gflags' \
'glog_0.3.4' \
'hdf5_1.8.17' \
'leveldb' \
'lmdb_0.9.18' \
'protobuf_3.0.0' \
'snappy_1.1.4' \
'cudnn_5.0' \
'opencv_2.4.13' \
'nccl_1.3.4' \
)

PREFIX_PATH="$HOME/local/cudnn:/usr/local/cuda"

for s in "${PACKAGES[@]}"; do
PREFIX_PATH=$PREFIX_PATH:$LOCAL/$s
done

CMAKE_PREFIX_PATH=$PREFIX_PATH cmake \
-DCUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda \
-DCMAKE_INSTALL_PREFIX=$LOCAL/caffe \
-DUSE_NCCL=ON \
.. | tee configure.log

make -j 248 all && make -j 248 test && make install
Clone this wiki locally