conditional-flow-matching icon indicating copy to clipboard operation
conditional-flow-matching copied to clipboard

Published 20 hours ago • verified publisher icon atong01

GPU Memory Allocation Issue with POT==0.9.1

Open atong01 opened this issue 1 year ago • 2 comments

With POT 0.9.1 GPU memory is allocated on import in combination with deep learning packages see https://github.com/PythonOT/POT/issues/523 https://github.com/PythonOT/POT/issues/516

Recommend downgrading to POT < 0.9.1 for now.

atong01 avatar Sep 16 '23 15:09 atong01

I just verified that this is fixed by https://github.com/PythonOT/POT/pull/520 when I install it from their master. Previously nearly all GPU memory would be all pre-allocated and now I can see the true usage.

RoyJames avatar Nov 03 '23 21:11 RoyJames

Oh that’s interesting! Thank you for your interest and checking this issue with us. We have been busy with the 1.0.4 release and we have not checked it for a while. @atong01 would you mind checking as well on the code that was broken?

kilianFatras avatar Nov 03 '23 23:11 kilianFatras