如何解决如何在 MXNet 中设置张量 dtypes = int64 的默认值？
现在我正在尝试使用 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 举报，一经查实，本站将立刻删除。
