I have successfully set up my Jetson Orin Nano Developer Super Kit and verified that PyTorch and CUDA are working. However, I am encountering the following error, and I can’t figure out the cause.
Device Information
JetPack Version : 6.2
PyTorch Version : 2.6.0+cu126
CUDA Version : 12.6
cuDNN Version : 90501
Python Code
import torch
print("PyTorch Version:", torch.__version__)
print("CUDA Version:", torch.version.cuda)
print("cuDNN Version:", torch.backends.cudnn.version())
print("CUDA Available:", torch.cuda.is_available())
if torch.cuda.is_available():
device = torch.device("cuda")
print("CUDA is available!")
print("Device Name:", torch.cuda.get_device_name(0))
x = torch.tensor([1.0, 2.0, 3.0]).to(device)
y = x * 2
print(y)
else:
print("GPU is not available, running on CPU.")
Output
PyTorch Version: 2.6.0+cu126
CUDA Version: 12.6
cuDNN Version: 90501
CUDA Available: True
CUDA is available!
Device Name: Orin
Traceback (most recent call last):
File ~/Desktop/Proje/untitled0.py, line 20
y = x * 2
RuntimeError: CUDA error: no kernel image is available for execution on the device
CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1
Compile with `TORCH_USE_CUDA_DSA` to enable device-side assertions.
Issue
RuntimeError: CUDA error: no kernel image is available for execution on the device
CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1
Compile with TORCH_USE_CUDA_DSA to enable device-side assertions.
How can I fix this issue? Do I need a custom PyTorch build for Jetson Orin Nano Super ?
However, from what I’ve read in other sources, it seems that 6.2 is not supported. I found that downgrading the package might be necessary depending on the JetPack version.
Hi,
I would like to mention that I don’t have prior experience with this topic. Could you kindly assist me by specifying what I should download from the link you provided?
No kernel image error indicates the package doesn’t build with the corresponding GPU architecture. For the Orin series, sm is 8.7.
This is a common issue on Jetson because most of the third-party packages are built for the dGPU device which has different architectures.
As a result, we provide some common packages for the Jetsons community in the link shared above.
After setting up your device, please download the package you need and install it.