-
Notifications
You must be signed in to change notification settings - Fork 499
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
ot.solve uses GPU even though tensors are on CPU? #612
Comments
Hello @mathurinm , It might relate to the following closed issue #516 , are you using POT >= 0.9.2 ? |
Thanks for the quick reply Cedric, this is happening with the latest dev version :
Contrary to #516, the memory consumption starts when calling a function from Let me know if I can provide additional info |
Ok thank you for the feedback, we will look into this and go back to you asap. |
Describe the bug
Running
ot.solve
with tensors on the CPU allows memory on the GPU (thisis documented inget_backend_list
) but also seems to use the GPU, as the Watts are increasing. See attached screencast :Screencast from 08-03-2024 11:24:44.webm
Is it normal?
Script
The text was updated successfully, but these errors were encountered: