更新时间:2024-05-23 GMT+08:00
如何解决训练过程中出现的cudaCheckError错误?
问题现象
Notebook中,运行训练代码出现如下错误。
cudaCheckError() failed : no kernel image is available for execution on the device
原因分析
因为编译的时候需要设置setup.py中编译的参数arch和code和电脑的显卡匹配。
解决方法
对于GP Vnt1的显卡,GPU算力为-gencode arch=compute_70,code=[sm_70,compute_70],设置setup.py中的编译参数即可解决。
父主题: 代码运行故障