AI开发平台ModelArtsAI开发平台ModelArts

文档首页> AI开发平台ModelArts> 常见问题> Notebook> 代码运行常见错误> 如何解决训练过程中出现的cudaCheckError错误?
更新时间:2021/03/18 GMT+08:00
分享

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

问题现象

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中的编译参数即可解决。

分享:

代码运行常见错误 所有常见问题

more