运行时错误:CUDA 错误:没有可在设备上执行的内核映像

时间:2021-07-08 21:14:39

标签: pytorch

我目前正在从事 MAXIM 集成 AI 项目。有关更多信息,请查看此 Github 链接: https://github.com/MaximIntegratedAI/ai8x-synthesis 我一直在按照 README 文件中提到的步骤进行操作,并到达了模型训练和量化部分。为了在 MAX78000 上为 MNIST 训练 FP32 模型,我运行了以下命令: bash 脚本/train_mnist.sh

但我不断收到错误消息: 运行时错误:CUDA 错误:没有可在设备上执行的内核映像

任何解决问题的指导或提示都会有很大帮助。谢谢!

0 个答案:

没有答案