gpt4 book ai didi

java - 有没有办法为 TensorFlow 进程设置 JAVA_OPTS

转载 作者:可可西里 更新时间:2023-11-01 15:50:09 25 4
gpt4 key购买 nike

现在我们使用 TensorFlow 将模型加载到 HDFS 中。该模型大于 1G,并且在达到堆大小时抛出 OOM。

2018-05-21 13:26:34.339483: I tensorflow/core/platform/cpu_feature_guard.cc:140] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 FMA
18/05/21 13:26:35 WARN util.NativeCodeLoader: Unable to load native-hadoop library for your platform... using builtin-java classes where applicable
INFO:tensorflow:Restoring parameters from hdfs://namenode:port/saved_model/0/variables/variables
INFO:tensorflow:Restoring parameters from hdfs://namenode:port/saved_model/0/variables/variables
2018-05-21 13:26:40.479367: W tensorflow/core/framework/allocator.cc:101] Allocation of 411041792 exceeds 10% of system memory.
hdfsPread: NewByteArray error:
java.lang.OutOfMemoryError: Java heap space

TensorFlow 现在正在使用 libhdfs,它可能会公开一些使用参数或环境变量设置 JAVA_OPTS 的方法。

最佳答案

最后,这将适用于 libhdfs:export LIBHDFS_OPTS="-Xms2048m -Xmx2048m"

关于java - 有没有办法为 TensorFlow 进程设置 JAVA_OPTS,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/50583056/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com