使用SageMaker v2.29.2和Tensorflow v2.3.2,我正在尝试实现分布式训练,如下面的博客文章所述:
然而,我在导入smdistributed脚本时遇到了困难。
这是我的代码:
我错过了什么?
这是我的代码:
import tensorflow as tf
import smdistributed.modelparallel.tensorflow as smp
错误:
Traceback (most recent call last):
File "temp.py", line 2, in <module>
import smdistributed.modelparallel.tensorflow as smp
ModuleNotFoundError: No module named 'smdistributed'
我错过了什么?
Tensorflow 2.3.1
或2.4.1
,并告诉我们是否解决了您的问题? - user11530462