首页 > 解决方案 > 限制单个 GPU 上 tensorflow 的资源

问题描述

我使用 RTX2080 GPU 在本地机器上运行一些tensorflow.keras代码(使用官方 tensorflow docker 映像)。由于内存和 GPU-Util 几乎用尽,机器的响应变得非常缓慢。我仍然希望能够在那台机器上工作,那么有没有办法限制/限制资源(类似于nice -n10 ...在 unix 上)?

提前致谢!

标签: dockertensorflowkerasgpu

解决方案


推荐阅读