TensorFlow 2.0 - tf.distribute 分布式训练
文章目录1. 单机多卡 MirroredStrategy2. 多机训练 MultiWorkerMirroredStrategy学习于:简单粗暴 TensorFlow 21. 单机多卡 MirroredStrategy# 分布式训练import tensorflow as tfimport tensorflow_datasets as tfds# 1 单机多卡 MirroredStrategystrategy = tf.distribute.MirroredStrategy()# 指定设备