I’m running with the following versions:
windows 10 cuda 11.6 cudnn 8.4 python 3.8.10
Those versions works when I’m running
torch models (I can run
torch models on GPU !).
I’m trying to install and run
When I installed
tensorflow 2.4.0 or
2.4.1 the GPU is not detected via:
import tensorflow physical_devices = tensorflow.config.list_physical_devices('GPU')
No GPU is detected (
physical_devices is empty)
When I upgraded the tensorflow version to 2.8, the GPU is detected (
physical_devices is not empty)
I’m run the following command before building and fitting my
import tensorflow physical_devices = tensorflow.config.list_physical_devices('GPU') tensorflow.config.experimental.set_memoty_growth(physical-devices, enable=True)
But I’m getting error:
Failed to find the dnn implementation
- I can’t change the
CUDA & CUDNNversions because they works for my
torchmodels (which used GPU and it works fine there).
What do I need to do in order to work with
tensorflow with GPU in my machine ?