Turoad/CLRNet

CUDA error

Opened this issue · 1 comments

Hi, I get the following error:

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

I suppose the error arises from this problem(?):

/home2/valentina.miller/.local/lib/python3.8/site-packages/torch/cuda/init.py:104: UserWarning:
NVIDIA GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70.
If you want to use the NVIDIA GeForce RTX 3090 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

How can I solve the incompatibility of the cuda and the current pytorch installation?
As pytorch version 1.8.0 is required....

Or is the origin of the error something else?

Any help would be very much appreciated, thanks!

Hi Valentina,
if you're using CUDA 10.x maybe you could try upgrading to CUDA 11
check discussion here