oneflow icon indicating copy to clipboard operation
oneflow copied to clipboard

flow.cuda.manual_seed_all() 会占用所有显卡的部分显存

Open BBuf opened this issue 3 years ago • 1 comments

import oneflow as flow

flow.cuda.manual_seed_all(1)

程序调用了flow.cuda.manual_seed_all之后会立刻占用每一张显卡的部分显存,我这里每张卡都被占用了500-600MiB,可能是异常现象。pytorch调用这个接口并没有观察到会占用显卡的显存。

BBuf avatar Nov 07 '22 07:11 BBuf

这个几百兆应该是cuda context占用的,这个现象原因应该是oneflow和pytorch创建cuda context的时机区别造成的。 oneflow应该是在此处直接初始化cuda了,而pytorch应该是存起来延迟初始化,导致的。

shangguanshiyuan avatar Nov 07 '22 09:11 shangguanshiyuan