search

ufoym / deepo

Setup and customize deep learning environment in seconds.
http://ufoym.com/deepo
MIT License
6.32k stars 749 forks source link

Problem with cudnn #149

Closed nagi1999a closed 2 years ago

nagi1999a commented 2 years ago

I am using the image tag all-py36-cu111 to create LSTM model with Keras. However, I find that it will cause the following error:

Fail to find the dnn implementation.

And cause the following error in the console:

Loaded runtime CuDNN library: 8.0.5 but source was compiled with: 8.1.0

From the base image nvidia/cuda:11.1-cudnn8-devel-ubuntu18.04, I find that the cudnn version 8.0.5.39 is too old for the latest tensorflow-gpu package. Due to this, the base image maybe need to change to 11.2.2-cudnn8-devel-ubuntu18.04?

ufoym commented 2 years ago

Fixed 38c9607b06f089a9d3eea0455c39382b91cae1bd