问题描述
现在我正在尝试使用 MXNet 包训练模型，但它引发了张量数据类型的问题。信息如下：
Version:
mxnet 1.8.0
mxnet-cu110 1.8.0
CUDA 11.0
cuDNN 8.0.4
错误如下
MXNetError: Check Failed: inp->shape().Size() < (int64_t{1} << 31) - 1 (15458446224 vs. 2147483647) : [SetNDInputsOutputs] Size of tensor you are trying to allocate is larger than 2^31 elements. Please build with flag USE_INT64_TENSOR_SIZE=1
如何解决这个问题？正如错误所说，使用标志构建，但我没有看到有关如何使用标志构建 mxnet-cuda 的任何具体指南。有没有其他方法可以在代码开始时设置默认的 dtype='int64'？
非常感谢！
解决方法
暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！
如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@）