On this page
Help Center/ ModelArts/ Troubleshooting/ DevEnviron/ Code Running Failures/ What Do I Do If cudaCheckError Occurs During Training?

What Do I Do If cudaCheckError Occurs During Training?

Updated on 2024-06-11 GMT+08:00

Symptom

The following error occurs when the training code is executed in a notebook:

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

Possible Cause

Parameters arch and code in setup.py have not been set to match the GPU compute power.

Solution

For Tesla V100 GPUs, the GPU compute power is -gencode arch=compute_70,code=[sm_70,compute_70]. Set the compilation parameters in setup.py accordingly.

Feedback

Feedback

Feedback

0/500

Selected Content

Submit selected content with the feedback