search

VoxelCubes / PanelCleaner

An AI-powered tool to clean manga panels.
GNU General Public License v3.0
208 stars 16 forks source link

I can't run on cuda #22

Closed Henicked closed 11 months ago

Henicked commented 11 months ago

I have the newest cuda installed 12.2. My gpu is supported. But seem like I can't run on cuda. Please help me.

VoxelCubes commented 11 months ago

That's unfortunate, and unfortunately I'm no expert on the matter either. The best I can do is point you towards this stackoverflow question: https://stackoverflow.com/questions/60987997/why-torch-cuda-is-available-returns-false-even-after-installing-pytorch-with

It presents multiple ways to diagnose and solve the issue. It's probably caused by your driver not getting along with the cuda interface the torch package expects, or the torch package wasn't installed with the cuda module, for some reason. That's probably what's going on. Good luck!

VoxelCubes commented 11 months ago

I found a workaround that should work:

pip uninstall torch torchvision -y
pip install torch torchvision --index-url https://download.pytorch.org/whl/cu118

Otherwise, try one of the pre-built binaries I'll be releasing over the next week.