文档首页 > > 常见问题> 开发环境> Notebook>

如何解决训练过程中出现的cudaCheckError错误?

如何解决训练过程中出现的cudaCheckError错误?

分享
更新时间:2020/12/29 GMT+08:00

问题现象

Notebook中,运行训练代码出现如下错误。

cudaCheckError() failed : no kernel image is available for execution on the device

原因分析

因为编译的时候需要设置setup.py中编译的参数arch和code和电脑的显卡匹配。

解决方法

对于Tesla V100的显卡,GPU算力为-gencode arch=compute_70,code=[sm_70,compute_70],设置setup.py中的编译参数即可解决。

分享:

文档是否有解决您的问题?

提交成功!非常感谢您的反馈,我们会继续努力做到更好!
反馈提交失败,请稍后再试!

*必选

请至少选择或填写一项反馈信息

字符长度不能超过200

提交反馈 取消

如您有其它疑问,您也可以通过华为云社区问答频道来与我们联系探讨

智能客服提问云社区提问