RuntimeError: CUDA error: no kernel image is available for execution on the device

[ashuai1114]

Hello, when I was trying to reproduce your code and run the model building module in the example.ipynb, I encountered the following error. However, I am using the Docker image provided by you. Could you please advise if there is a solution for this?

RuntimeError: CUDA error: no kernel image is available for execution on the device

[KonstantinWilleke]

Hi, could you post the entire error message here? There are probably issues with the installation of the cuda drivers.
Could you run torch.cuda.is_available() in the environment? if it returns false, then you have to make sure that you have a nvidia GPU in your machine, and a cuda version of 12.0 or higher.