search

Cysu / dgd_person_reid

Domain Guided Dropout for Person Re-identification
http://arxiv.org/abs/1604.07528
231 stars 94 forks source link

Can I only using one GPU #3

Closed duanLH closed 8 years ago

duanLH commented 8 years ago

I only have one GPU ， so I do not want to use openmpi ,if without openmpi ,can I using this project sucessfully ?

Cysu commented 8 years ago

Sure. You can safely comment out the lines of USE_MPI, MPI_INCLUDE, and MPI_LIB in Makefile.config. But also remember to change these lines to

GLOG_logtostderr=1 ${CAFFE_DIR}/build/tools/caffe train \
  -solver ${solver} -gpu 0 2>&1 | tee ${log}

and 

GLOG_logtostderr=1 ${CAFFE_DIR}/build/tools/caffe train \
  -solver ${solver} -weights ${pretrained_model} -gpu 0 2>&1 | tee ${log}