Модель поезда Tensorflow 2.0 на одном графическом процессоре

Question

Я хочу обучить модель последовательного тензорного потока (версия 2.3.0) на одной карте NVIDIA graphi c (RTX 2080 super). Я использую следующий фрагмент кода для создания и обучения модели. Однако каждый раз, когда я запускаю этот код, я не вижу использования графического процессора. Есть предложения, как изменить мой код, чтобы я мог запускать его на 1 графическом процессоре?

strategy = tf.distribute.OneDeviceStrategy(device="/GPU:0")
with strategy.scope():
    num_classes=len(pd.unique(cats.No))
    model = BuildModel((image_height, image_width, 3), num_classes)
    model.summary()
    model=train_model(model,valid_generator,train_generator,EPOCHS,BATCH_SIZE)

Answer 1

запустите приведенный ниже код, чтобы узнать, обнаруживает ли тензорный поток ваш графический процессор.

import tensorflow as tf
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
print(tf.__version__)
print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))
tf.test.is_gpu_available()
!python --version