Versions of cuda and cudnn

ronghanghu / snmn

Code release for Hu et al., Explainable Neural Computation via Stack Neural Module Networks. in ECCV, 2018

http://ronghanghu.com/snmn/

BSD 2-Clause "Simplified" License

72 stars 6 forks source link

Open yash-iiith opened 5 years ago

yash-iiith commented 5 years ago

I request you to please mention the versions of cuda and cudnn used because I am facing some errors concerning cuda and cudnn.

ronghanghu commented 5 years ago

Hi Yash! I used cuDNN 7.1 and CUDA 9.0 in my experiments.

Regarding the cuDNN and CUDA issues, here is a list of cuDNN and CUDA version for each pre-build TensorFlow package.