keras ValueError:错误的封送数据(未知类型代码)



我想测试从另一台计算机创建的模型时遇到了困难。

我运行模型的代码是:

import tensorflow.keras.models as M
org_net = M.load_model(route_net, compile=True, custom_objects={'B': keras.backend,'diceloss':diceloss}) 

在这里我得到了一个错误,当我在其他电脑上尝试时,我看不到

File "C:UsersEuijeong SongPycharmProjectsdl_nxbvenvtest_org.py", line 30, in <module>
org_net = M.load_model(route_net, compile=True, custom_objects={'B': keras.backend,'diceloss':diceloss})#
File "C:UsersEuijeong SongPycharmProjectsdl_nxbvenvlibsite-packageskerasutilstraceback_utils.py", line 67, in error_handler
raise e.with_traceback(filtered_tb) from None
File "C:UsersEuijeong SongPycharmProjectsdl_nxbvenvlibsite-packageskerasutilsgeneric_utils.py", line 792, in func_load
code = marshal.loads(raw_code)
ValueError: bad marshal data (unknown type code)

我的python版本是3.9.0,tf==2.7.0。并创建了python==3.6.8和tf==2.1.0的模型。我发现在使用python=2.x.x时会出现封送错误,但这次我不明白。为什么会发生这种情况,我该如何解决?

我解决了这个问题。这是因为python和cudnn的版本太高了。

我将cudnn降级为8.2.2,这适用于cuda 11.4,将python降级为3.7.0

相关内容

  • 没有找到相关文章

最新更新