之前貌似用的是tensorflow2.2.0跑的，反正用gpu跑的很成功 大约四个小时就能跑出来，但是后来就崩了，但是他崩把 也不告诉你 就是跑不出个结果 我觉得也不是他自己的问题 是我自己想把tensorflow跑出来的结果转成在web端上的那种冻结文件 结果转版本把环境给转坏了
后来又试了tensorflow2.6.0 结果训练需要16个小时，把我人搞蒙了，我一看这没用gpu训练啊，反正我后面试了好几次 就是cuda11不行 我这边跑着跑着就崩 python就abort啥玩意的 我就换了tensorflow2.3.0 对应的是cuda的10 虽然不崩了把 但是报其他的错 啥layer啥的 我就更换了cudnn的版本 因为cuda10 对应的cudnn 有7.4 7.6 7.1之类的，都是cudnn7 换了个最高的版本7.6就跑出来
写出来给大家借鉴一下 又不会的在底下问哈
版权声明：本文为CSDN博主「ysera_wang」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/vickieyy/article/details/121630472
暂无评论