Closed alon1samuel closed 5 years ago
Hello @alon1samuel
Did you manage to find a fix for the above error by any chance? Sorry I have been trying to get it working for the last one week but have not been having too much luck.
Thanks, Vignesh
Hi, I don't remember if I fixed it or how. From reading the message that it gave me, my guess is that the main part is "cudnn failed to initialize". I would guess that it's not a problem with this repo, but a problem with any TF model you are running that uses conv2d layers (or similar). So I would suggest to check back the installation again with a simple model like in this tutorial to see it works first. Hope it helps! Alonsh
Hi, thanks for your reply. I got it working a similar way to how you did initially by limiting GPU usage. I shall try it on the simple model and try to figure out the problem.
Thanks
Hi,
I've tried to make demo run and I got this error:
I've make it work with inserting lines according to GPU memory allocation in this issue.
Does anyone has a better solution or is that it?
Regards, Alon