gpu-limit.md

@@ -15,9 +15,8 @@ import tensorflow as tf
 
 gpus = tf.config.experimental.list_physical_devices('GPU')
 if gpus:
-    tf.config.experimental.set_virtual_device_configuration(
-      gpus[0], [tf.config.experimental.VirtualDeviceConfiguration(memory_limit=megabytes)]
-    )
+    for gpu in gpus:
+        tf.config.experimental.set_virtual_device_configuration(gpu [tf.config.experimental.VirtualDeviceConfiguration(memory_limit=megabytes)])
 ```