Есть ли способ установить JAVA_OPTS для процесса TensorFlow - PullRequest
0 голосов
/ 29 мая 2018

Теперь мы используем 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 с параметрами или переменными среды.

1 Ответ

0 голосов
/ 31 мая 2018

Наконец, это будет работать для libhdfs: export LIBHDFS_OPTS="-Xms2048m -Xmx2048m".

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...