问题描述:
【操作步骤&问题现象】
1、执行model.train报错
【截图信息】
解答:
是用的GPU训练的吗?这个是报的多卡的错误,可以通过设置GPU的哪张卡:
from mindspore import context
context.set_context(device_target='GPU', device_id=0)
RuntimeError: CUDNN_STATUS_EXECUTION_FAILED |【❤️有效解决❤️】-云社区-华为云
问题描述:
【操作步骤&问题现象】
1、执行model.train报错
【截图信息】
解答:
是用的GPU训练的吗?这个是报的多卡的错误,可以通过设置GPU的哪张卡:
from mindspore import context
context.set_context(device_target='GPU', device_id=0)
RuntimeError: CUDNN_STATUS_EXECUTION_FAILED |【❤️有效解决❤️】-云社区-华为云