tensorflow:19.12-tf2-py3 no multiple gpus

AI & Data Science Deep Learning (Training & Inference) Frameworks
1

Hi everyone,

I am trying to run the GNC container tensorflow:19.12-tf2-py3. It

  1. can detect two GPUs
  2. When I run without tensorflow strategy = tf.distribute.MirroredStrategy(), it is OK
  3. When strategy = tf.distribute.MirroredStrategy(), it brings error: distributed_function
  4. Without GNC container, the code is good with multple gpus

Any suggestions?
Thank you!