Closed Flionay closed 2 years ago
The code about muti-gpus should be:
physical_devices = tf.config.experimental.list_physical_devices('GPU') if len(physical_devices) > 0: tf.config.set_visible_devices(physical_devices[0:1], 'GPU')
The code about muti-gpus should be: