Skip to content

Instantly share code, notes, and snippets.

@trongan93

trongan93/ship_detection_physical_memory

Created June 12, 2020 08:43
Show Gist options
  • Save trongan93/719ed503a5e9c081fbf88bf43630c9d5 to your computer and use it in GitHub Desktop.
Save trongan93/719ed503a5e9c081fbf88bf43630c9d5 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
ship_detection_physical_memory
physical_devices = tf.config.experimental.list_physical_devices('GPU')
for physical_device in physical_devices:
tf.config.experimental.set_memory_growth(physical_device, True)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment