1024 MB should be enough for anyone

old-commit-hash: 873043b9fb
commatwo_master
Willem Melching 5 years ago
parent b41a7bd1dc
commit e4e26ca57e
  1. 3
      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)

Loading…
Cancel
Save