#4040·horovod

是否要将 tf.train.SessionRunHook 替换为 tf.compat.v1.train.SessionRunHook?

作者: whatdhack创建于 2024年5月1日更新于 2024年10月8日
标签bug

环境:

  1. 框架: TensorFlow,
  2. 框架版本: 2.16
  3. Horovod 版本: 0.28.1
  4. MPI 版本:
  5. CUDA 版本: 12.2
  6. NCCL 版本:
  7. Python 版本: 3.11.8
  8. Spark/PySpark 版本:
  9. Ray 版本:
  10. 操作系统和版本:
  11. GCC 版本:
  12. CMake 版本:

内容来源: horovod/horovod