setup problem
123liluky opened this issue · 2 comments
Thanks for your job. Must we install tf1.6 instead of any other versions? And what version of cuda and cudnn should I install? I have installed tf1.4, cuda8.0 and cudnn6.
Hi @123liluky , thank you for your interest in the repo.
No, I confirmed this model works with tf1.5, tf1.6, and tf1.7. But regarding training performance ft1.6 was best. As you may refer to here, I was able to use larger batch size with tf1.6 than tf1.5. For some reason, performance tf1.7 is worse than tf1.6. Sorry I never check if it work with tf1.4.
Regarding cuda and cudnn, I confirm that the model works with cuda9.0 and 9.1, and cudnn 7 and 7.1.
couldn’t find 'checkpoint' file or checkpoints in given directiory @ @ @123liluky @rishizek