CODEWITHSUNDEEP
pythontensorflowkeras
h_llmnt
h_llmnt

Reputation: 11

Training CNN model using keras works with CPU but not with GPU

I am training a CNN model to classify simple images (squares and crosses) and everything works just fine when I use the cpu but when I use the gpu everything works until the training starts and i get this error:

2022-06-15 04:25:49.158944: I tensorflow/stream_executor/cuda/cuda_dnn.cc:384] Loaded cuDNN version 8401

And then the program just stops. Does anyone have an idea how to fix this?

Upvotes: 1

Views: 261

Answers (1)

Ziv Yong
Ziv Yong

Reputation: 21

if you use pycharm, you can select the "Emulate terminal in output console" option to print detailed error information.

Run->Edit Configration->Execution ->Emulate terminal in output console

On windows, maybe CUDA is missing zlibwapi.dll file, and you can download it and move it to bin of cuda.

Upvotes: 2

Related Questions