java - 内存不足错误: Java heap space on Tensorflow tests execution

Question

使用命令执行 Tensorflow 测试套件:

bazel test //tensorflow/...

抛出内存不足问题:

INFO: Found 1886 targets and 1155 test targets...
INFO: Elapsed time: 291.775s, Critical Path: 0.91s
java.lang.OutOfMemoryError: Java heap space
        at com.google.devtools.build.skyframe.SkyKey.create(SkyKey.java:57)
        at com.google.devtools.build.lib.skyframe.ArtifactSkyKey.key(ArtifactSkyKey.java:43)
        at com.google.devtools.build.lib.skyframe.ActionExecutionFunction.toKeys(ActionExecutionFunction.java:576)
        at com.google.devtools.build.lib.skyframe.ActionExecutionFunction.compute(ActionExecutionFunction.java:158)
        at com.google.devtools.build.skyframe.ParallelEvaluator$Evaluate.run(ParallelEvaluator.java:370)
        at com.google.devtools.build.lib.concurrent.AbstractQueueVisitor$WrappedRunnable.run(AbstractQueueVisitor.java:501)
        at java.util.concurrent.ThreadPoolExecutor.runWorker(ThreadPoolExecutor.java:1142)
        at java.util.concurrent.ThreadPoolExecutor$Worker.run(ThreadPoolExecutor.java:617)
        at java.lang.Thread.run(Thread.java:745)
Java heap space

bazel ran out of memory and crashed.

我已尝试以下选项，问题仍然存在。

export _JAVA_OPTIONS="-Xms1024m -Xmx1024m"

export JVM_ARGS="-Xmx1024m -XX:MaxPermSize=256m"

export JVM_ARGS="-XX:PermSize=64M -XX:MaxPermSize=256m"

此外，还更改了 bazel 代码中的文件:scripts/bootstrap/compile.sh

"运行 "${JAVAC}"-J-Xms1g -J-Xmx1g -classpath "${classpath}"-sourcepath "${sourcepath}""

机器配置:Ubuntu发行版，openjdk8，RAM 16G

Answer 1

尝试使用正确的名称设置JAVA_OPTIONS并且不带引号字符:

JAVA_OPTIONS= -Xms512m -Xmx1024m

如果不起作用，可能是没有使用 JAVA_OPTIONS 环境变量。
您可以尝试直接在执行命令中指定参数。 According to the official documentation ，你可以这样做:

bazel test --host_jvm_args=-Xms512m --host_jvm_args=-Xmx1024m //tensorflow/...