fix #12 issue about cudnn handler on RTX graphics cards

arnab39 / FewShot_GAN-Unet3D

Tensorflow implementation of our paper: Few-shot 3D Multi-modal Medical Image Segmentation using Generative Adversarial Learning

MIT License

377 stars 82 forks source link

fix #12 issue about cudnn handler on RTX graphics cards #14

Closed psicobloc closed 2 years ago

psicobloc commented 4 years ago

added allow_growth = FLAGS.gpu_growth as part of the gpu_options and I set such flag to True, this resolves #12 This specific issue is also on the Tensorflow repo (https://github.com/tensorflow/tensorflow/issues/24496) but has been solved in the comments, the issue is still open.