1024 MB should be enough for anyone

5 years ago · 873043b9fb
parent 9ad9076486
commit 873043b9fb
1 changed files with 1 additions and 2 deletions
--- a/selfdrive/modeld/runners/keras_runner.py
+++ b/selfdrive/modeld/runners/keras_runner.py
@ -41,10 +41,9 @@ if __name__ == "__main__":
  # limit gram alloc
  gpus = tf.config.experimental.list_physical_devices('GPU')
  if len(gpus) > 0:
-    tf.config.experimental.set_virtual_device_configuration(gpus[0], [tf.config.experimental.VirtualDeviceConfiguration(memory_limit=2048)])
+    tf.config.experimental.set_virtual_device_configuration(gpus[0], [tf.config.experimental.VirtualDeviceConfiguration(memory_limit=1024)])

  m = load_model(sys.argv[1])
  print(m, file=sys.stderr)

  run_loop(m)
-