search

Mazhar004 / MODNet-BGRemover

A deep learning approach to remove background & adding new background image
120 stars 59 forks source link

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

Open finnishbroccoli opened 1 year ago

finnishbroccoli commented 1 year ago

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 commented 1 year ago