Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

AssertionError: Torch not compiled with CUDA enabled #35

Closed
ftk789 opened this issue Apr 8, 2022 · 4 comments
Closed

AssertionError: Torch not compiled with CUDA enabled #35

ftk789 opened this issue Apr 8, 2022 · 4 comments

Comments

@ftk789
Copy link

ftk789 commented Apr 8, 2022

image

@Sanster
Copy link
Owner

Sanster commented Apr 8, 2022

The torch version you installed does not support gpu, you can use lama cleaner with cpu: python3 main.py --device=cpu

@Sanster Sanster closed this as completed Apr 8, 2022
@ftk789
Copy link
Author

ftk789 commented Apr 9, 2022

Thank you <3 , Seems like the Torch version i had was not for GPU'S as you said

@Red007Master
Copy link

Thank you <3 , Seems like the Torch version i had was not for GPU'S as you said

how did you install CUDA compatible Torch?

@Jeslandor
Copy link

Jeslandor commented Oct 2, 2022

how did you install CUDA compatible Torch?

Just in case someone else had the same problem I was able to solve it by uninstalling torch and installing the cuda version with the next steps:

  1. pip uninstall torch
  2. pip install "torch==1.11.0+cu113" --index https://download.pytorch.org/whl/cu113/

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

4 participants