tensorflow 运行时错误：资源耗尽：试图分配553305856字节时内存不足,BufferAssignment OOM

0qx6xfy6 于 2022-11-25 发布在其他

关注(0)|答案(1)|浏览(313)

我得到这个错误时，在多个GPU上运行一个jax脚本.

jaxlib.xla_extension.XlaRuntimeError: RESOURCE_EXHAUSTED: Out of memory while trying to allocate 553305856 bytes. BufferAssignment OOM

我能做些什么来解决这个问题吗？

tensorflow

来源：https://stackoverflow.com/questions/74143812/jaxlib-xla-extension-xlaruntimeerror-resource-exhausted-out-of-memory-while-tr

1条答案

按热度按时间

zi8p0yeb1#

这似乎对我起了作用。

os.environ["XLA_PYTHON_CLIENT_PREALLOCATE"]="false"
os.environ["XLA_PYTHON_CLIENT_MEM_FRACTION"]=".XX"
os.environ["XLA_PYTHON_CLIENT_ALLOCATOR"]="platform"

https://jax.readthedocs.io/en/latest/gpu_memory_allocation.html

赞(0）回复(0）举报 2022-11-25

我来回答

tensorflow 运行时错误：资源耗尽：试图分配553305856字节时内存不足,BufferAssignment OOM

1条答案

相关问题

热门标签

最新问答