search

USTC-JialunPeng / Diverse-Structure-Inpainting

CVPR 2021: "Generating Diverse Structure for Image Inpainting With Hierarchical VQ-VAE"
MIT License
175 stars 19 forks source link

The problem that GPU(A4000) does not adapt to tf1.12 version #20

Open plusleft opened 2 years ago

plusleft commented 2 years ago

The paper is exciting! Thanks to the authors. The program runs perfectly on the GPUs(two 2080), but the same version(TensorFlow-gpu=1.12.0 cuda=9.0 cudnn=7.6.5) cannot be adapted to the two A4000. Does anyone have a solution to the version problem?

Austin-111 commented 2 years ago

Thanks for your interest in our work! Maybe you can try another TensorFlow 1.x version (lower or higher than 1.12) which is compatible with A4000. But I don't know if this version exists.

plusleft commented 2 years ago

@Austin-111 Thank you for your reply! I have tried many versions, but none of them can be directly ported. If the version difference is too big, there may be a lot of code that needs to be changed. If you have not tried other GPU, then I could try to modify the code to fit the version.

USTC-JialunPeng commented 2 years ago

I remember we have tried our code on GTX 1080Ti. It seems some modifications are needed if the version difference is big. Good luck!