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

[finnishbroccoli]

I've installed BGRemover to Linux. I had some problems with torchvision version, because pip did not understand requirements.txt for some reason.

I have the following packages installed

Package Version

---

numpy 1.24.1 opencv-python 4.5.1.48 Pillow 6.2.2 pip 22.3.1 setuptools 65.6.3 six 1.16.0 torch 1.7.1 torchvision 0.2.2.post3 typing_extensions 4.4.0

When I run python inference.py --image

I get "Erro happend CUDA error: no kernel image is available for execution on the device"

So, should I modify torchvision version, install some additional pip package, both or something else?

[finnishbroccoli]

• I have CUDA version 12.0