MXNet с использованием NDArray Scatter_nd на нескольких графических процессорах

Question

Когда я использую multi gpu для обучения в MXNet (CUDA8.0 + cudnn7), я сначала инициализирую параметры в другом контексте, затем выполняю scatter_nd в разных контекстах, в первый раз scatter nd будет работать отлично, но при вычислении длявторая карта GPU, я получил

F1217 23: 53: 01.012707 2619 stream_gpu-inl.h: 62] Проверка не удалась: e == cudaSuccess CUDA: обнаружен недопустимый доступ к памяти

Answer 1

Это произойдет, если вы не скопировали свои данные во всех контекстах.В соответствии с github.com/apache/incubator-mxnet/issues/10240 это может быть исправлено с помощью github.com/apache/incubator-mxnet/pull/10833.Попробуйте с 1.3.1, если вы используете 1.3.0