aiff22 / DPED

Software and pre-trained models for automatic photo quality enhancement using Deep Convolutional Networks
1.68k stars 369 forks source link

Cudnn handle #22

Closed PhoenixLSQ closed 5 years ago

PhoenixLSQ commented 5 years ago

When i run the test code with GPU,I got the note as below:

Testing original sony model, processing image 17.jpg 2019-07-11 01:37:25.413701: E tensorflow/stream_executor/cuda/cuda_dnn.cc:353] Could not create cudnn handle: CUDNN_STATUS_NOT_INITIALIZED 2019-07-11 01:37:25.413830: E tensorflow/stream_executor/cuda/cuda_dnn.cc:361] Possibly insufficient driver version: 384.183.0 Segmentation fault (core dumped)

I use tensorflow-gpu 1.11.0, cuda9.0 I had change my cudnn version from 7.4.2 to 7.0.5. Both of these two version didn't work. I'll thank for any suggestion!

aiff22 commented 5 years ago

@PhoenixLSQ, it seems that your Nvidia Graphics Drivers drivers are out-of-date. You can download and install their latest version from Nvidia's official website: https://www.nvidia.com/Download/index.aspx