我按照你们给我的建议安装了tf1.15版本的镜像,报错,报错信息如下
代码中有写这个tf_config = tf.ConfigProto(allow_soft_placement=True)推理部分是用的你们教程上的代码,前处理和后处理都是仿造pipeline上写的,我不知道为什么用你们notebook就好用,为什么我本地下载了你们的镜像,代码也一样,怎么就是跑不起来
根据您提供的错误信息,问题出在TopKV2
节点的定义上。这个错误可能是由于TensorFlow版本不兼容导致的。您可以尝试以下方法解决这个问题:
pip install --upgrade tensorflow
pip install tensorflow==1.5
我用的模型卡片示例代码,在cpu上可以跑。https://modelscope.cn/models/damo/nlp_csanmt_translation_en2zh/summary 镜像:registry.cn-beijing.aliyuncs.com/modelscope-repo/modelscope:ubuntu20.04-py38-torch2.0.1-tf1.15.5-1.8.1。此回答整理自钉钉群:魔搭ModelScope开发者联盟群 ①