Closed thehzzz closed 2 years ago
目前还不支持显存按需调用,需要预先设定显存大小限制。 未来可能会支持动态的显存分配。
感谢回复。请问Ubuntu是通过什么方式去设定显存限制呢。
因为条件有限,希望在运行bminf的时候,留1G 左右显存。
cpm1 = bminf.models.CPM1(
dynamic_memory=512 * 1024 * 1024, # 512MB
memory_limit= 4 * 1024 * 1024 * 1024, # 4GB
)
非常感谢!
非常喜欢bminf。 请问是否支持显存按需调用。 非常感谢。