如何在kaggle中使用GPU使用TF进行预处理和模型训练？

Question

我有视频数据并需要提取帧（在 CPU 上完成大约 5 小时）。在 Kaggle 中，我可以使用 GPU 来完成此任务以及 CNN 模型训练。无法使用 TF 来使用 GPU。在 Pytorch 中我发现它很容易实现。

我试过了。 :

gpus = tf.config.experimental.list_physical_devices('GPU')
for gpu in gpus:
    print("Name:", gpu.name, "  Type:", gpu.device_type)

def prepare_all_videos(df, root_dir):
    device = '/device:GPU:0' if len(tf.config.list_physical_devices('GPU')) > 0 else '/device:CPU:0'
    num_samples = len(df)
    ...

Answer 1

我发现只有当库支持 GPU 加速时 GPU 才会被使用。例如：Numpy没有GPU加速，在kaggle中为它启用GPU，不起作用。