Problem of multi-GPU "'NCCL' has no attribute 'new_uid'" about py-r-fcn-multigpu HOT 5 CLOSED

change gpu mumber in the file rfcn_end2end_ohem_multi_gpu.sh ,the source code is 8 gpu
ime ./tools/train_net_multi_gpu.py --gpu ${GPU_ID}
--solver models/${PT_DIR}/${NET}/rfcn_end2end/solver_ohem.prototxt
--weights data/imagenet_models/${NET}-model.caffemodel
--imdb ${TRAIN_IMDB}
--iters ${ITERS}
--cfg experiments/cfgs/rfcn_end2end_ohem_${PT_DIR}.yml
${EXTRA_ARGS}

from py-r-fcn-multigpu.

HI Xiangyu, have you solved the problem? I have the same problem as you.

from py-r-fcn-multigpu.

@Xiangyu-CAS Have you solved the problem? I have the same problem as you.

from py-r-fcn-multigpu.

has anyone installed the caffe in this repo successfully?

from py-r-fcn-multigpu.

You have to export the path to the NCCL build. I made a local install so i exported it to my home
export LD_LIBRARY_PATH=/home/[username]/nccl/build/lib:$LD_LIBRARY_PATH

from py-r-fcn-multigpu.