-
Notifications
You must be signed in to change notification settings - Fork 537
Closed
Labels
Description
Describe the bug
I'm just using the import ot in my test.py and it causes all the cuda memory of my first GPU to be full
Screenshots
Environment (please complete the following information):
- OS (e.g. MacOS, Windows, Linux): Linux
- Python version: 3.10.13
- POT version: 0.9.1
- How was POT installed (source,
pip,conda): pip - Only for GPU related bugs:
- CUDA version: 12.0
Output of the following code snippet:
import ot
import pdb
pdb.set_trace()
print()