最近在跑这篇文章的代码时遇到了很多问题,特意记录下来,以供参考。
作者的github中给出了代码所需的运行环境,但是实验室没有空余的2080ti可以使用,只有3090ti可以使用,所以在改变版本时遇到了较多问题。
最终确定版本,RTX3090ti,python3.7,tensorflow2.5.0,keras2.3.1,CUDA11.1,cudnn8.1.0(注意CUDA和cudnn版本一定要特别注意!!!不然无法使用显卡运行程序,此搭配亲测可以)
在更改到tensorflow2.x时,代码中也有部分地方需要进行改动,
1、tensroflow版本问题
改为
这是由于tensorflow2.x相比tensorflow1.x减少了部分函数,此类问题较多,都可以通过这种方式进行解决。
2、运算问题
改为
这是因为tensorflow的版本问题。
在后续的运算中可以适当增加Batch_size的大小,加快训练速度。
使用3090ti,batch_size=8的训练速度大概在15小时左右。
后续问题欢迎与我讨论,本人邮箱595260329@qq.com