拉长雪碧

文章
6
资源
0
加入时间
2年10月18天

TensorFlow 2.0 - tf.distribute 分布式训练

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