mxnet mshadow/./stream_gpu-inl.h:62: Check failed: e == cudaSuccess: CUDA: an illegal memory access

最新推荐文章于 2022-07-25 20:50:29 发布

城俊BLOG

最新推荐文章于 2022-07-25 20:50:29 发布

阅读量771

点赞数

分类专栏： # DL-报错 MxNet Linux

本文链接：https://blog.csdn.net/qxqxqzzz/article/details/107757259

版权

Linux 同时被 3 个专栏收录

212 篇文章 30 订阅 ¥19.90 ¥99.00

订阅专栏

超级会员免费看

DL-报错

122 篇文章 12 订阅 ¥19.90 ¥99.00

订阅专栏

超级会员免费看

MxNet

43 篇文章 2 订阅

订阅专栏

在使用mxnet 1.6时遇到GPU错误：`mshadow/./stream_gpu-inl.h:62: Check failed: e == cudaSuccess: CUDA: an illegal memory access`。可能原因包括batch_size过大、ndarray context不一致或自定义OP问题。解决方案是确保所有运算涉及的ndarray在同一context，如mx.gpu(0)或mx.cpu()，若困难则可将计算转至CPU。

摘要由CSDN通过智能技术生成

mxnet 1.6

报错：

Traceback (most recent call last):
  File "train_0723.py", line 455, in <module>
    main()
  File "train_0723.py", line 451, in main
    train_net(args)
  File "train_0723.py", line 445,