PythonOT/POT

Cuda memory is full when import

zhangzef opened this issue · 1 comments

zhangzef commented

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

image

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()
rflamary commented

Yes this is a duplicate of #516 so I'm closing It. It shoudl be fixed in the master branch if you want to test it.