It's the problem of PinnedMemoryStorage: 
https://github.com/apache/incubator-mxnet/blob/master/src/storage/pinned_memory_storage.h#L59

I use the CUDA_VISIBLE_DEVICES environment variable to bypass this bug.

[ Full content available at: 
https://github.com/apache/incubator-mxnet/issues/10625 ]
This message was relayed via gitbox.apache.org for [email protected]

Reply via email to