search

rishizek / tensorflow-deeplab-v3

DeepLabv3 built in TensorFlow
MIT License
286 stars 102 forks source link

setup problem #9

Open 123liluky opened 6 years ago

123liluky commented 6 years ago

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.

rishizek commented 6 years ago

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.

XZNWU commented 6 years ago

couldn’t find 'checkpoint' file or checkpoints in given directiory @ @ @123liluky @rishizek