caffe2之deteron之multi-GPU training throw an illegal memory access

油炸小蚂蚁

于 2018-04-24 17:16:07 发布

阅读量357

点赞数

分类专栏： caffe2 caffe

本文链接：https://blog.csdn.net/yanhuangzhinu0/article/details/80067424

版权

caffe 同时被 2 个专栏收录

2 篇文章

订阅专栏

caffe2

1 篇文章

订阅专栏

一种建议是在python代码开始处写上

os.environ["CUDA_VISIBLE_DEVICES"]="0,1,2,3"

但本人尝试无效，同时也没有删除

加入了第二种方法：

python2 tools/train_net.py --multi-gpu-testing \
  --cfg configs/getting_started/tutorial_2gpu_e2e_faster_rcnn_R-50-FPN.yaml \
  OUTPUT_DIR /tmp/output USE_NCCL True

编译caffe2时要打开ncll，然后调用时 USE_NCCL True

本人最原始错误提示：[enforce fail at cudnn_wrappers.h:78] error == cudaSuccess. 77 vs 0.

大神讨论地址如下：

点击打开链接