Colab 未使用 GPU 進行培訓

Question

我正在 GTSRB 數據集上訓練自定義 model 用於 Google Colab 上的交通標志識別。 我已經成功構建了我的 model，但是當我訓練我的 model 時，它在 CPU 而不是 GPU 上運行。 我之前選擇了 GPU 運行時並使用 keras-Tensorflow 使用鏡像策略。 任何幫助將不勝感激。 我正在添加我的代碼片段。

with strategy.scope():

  test_datagen = ImageDataGenerator(rescale=1./255)
  train_datagen = ImageDataGenerator(rescale=1./255, shear_range=0.2, zoom_range=0.2, 
  horizontal_flip=False)
  train_generator = train_datagen.flow_from_directory('/content/drive/MyDrive/Train',target_size= 
  (64, 64),batch_size=32,class_mode='categorical')
  validation_generator = 
  test_datagen.flow_from_directory(directory='/content/drive/MyDrive/validation', target_size= 
  (64, 64), batch_size=32, class_mode='categorical')
  checkpoint_path = "training_1/cp.ckpt"
  cp_callback = ModelCheckpoint(filepath=checkpoint_path, save_weights_only=True, verbose=1, 
  save_freq='epoch')

with strategy.scope():
  model = training(classes)
  model.compile(loss='categorical_crossentropy', optimizer=Adam(lr=1e-4),metrics=['accuracy'])
model.fit(train_generator, steps_per_epoch=200, epochs=50, validation_data=validation_generator, 
validation_steps=100, callbacks=[cp_callback])

在訓練時，我檢查了 GPU 和 CPU 使用情況，如圖所示：

Answer 1

在 model 使用命令nvidia-smi運行時檢查 GPU 的狀態。

Answer 2

我之前在多個平台（Kaggle、Colab）上曾多次遇到過這種情況，但只設法解決了一次。 在那種情況下，這是由於在我切換 tensorflow 版本時未安裝 tensorflow_gpu（tensorflow 的 gpu 版本）。 您可以檢查 tensorflow_gpu 的版本和安裝的 tensorflow 的版本。 確保版本相同。 這可能完全不合時宜，也可能對您有用。