stu1130 opened a new issue #18834:
URL: https://github.com/apache/incubator-mxnet/issues/18834


   ## Step to reproduce
   * build mxnet-cu101 from source based off mxnet 1.7.0 branch
   * gluoncv 0.7
   ```
   from gluoncv import model_zoo, data, utils
   from matplotlib import pyplot as plt
   import mxnet as mx
   from mxnet import gluon 
   
   net = model_zoo.get_model('yolo3_darknet53_coco', pretrained=True, 
ctx=mx.gpu())
   net.hybridize()
   x = mx.nd.random.uniform(shape=(1, 3, 1000, 1000), ctx=mx.gpu())
   _, scores, _ = net(x)
   print(scores.shape)
   net.export("yolo")
   
   deserialized_net = gluon.nn.SymbolBlock.imports("yolo-symbol.json", 
['data'], "yolo-0000.params", ctx=mx.gpu())
   image = mx.nd.random.normal(shape=(1, 3, 1000, 1000), ctx=mx.gpu())
   print(deserialized_net(image))
   ```
   ```
   CUDA: Check failed: e == cudaSuccess: an illegal memory access was 
encountered
   [21:53:46] src/resource.cc:279: Ignore CUDA Error [21:53:46] 
src/storage/./pooled_storage_manager.h:97: CUDA: an illegal memory access was 
encountered
   
   
   [21:53:46] src/resource.cc:331: Ignore CUDA Error [21:53:46] 
src/common/random_generator.cu:70: Check failed: e == cudaSuccess || e == 
cudaErrorCudartUnloading: CUDA: an illegal memory access was encountered
   ```
   It works fine when I use mxnet-cu101 1.6 pip wheel
   


----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

For queries about this service, please contact Infrastructure at:
[email protected]


Reply via email to