cpu?
Или гпу недоступен, или в коде надо указывать
ПРоверьте доступность GPU import tensorflow as tf tf.config.list_physical_devices('GPU')
для старой версии print("GPU Available: ", tf.test.is_gpu_available())
В этом образе в юпитере работает: https://hub.docker.com/r/yhavinga/jupyter-tensorflow-pytorch-gpu
Проверить версию драйверов, версию тф и куду
нужны ещё cudnn, cuda
Обсуждают сегодня