Tensorflow gpu未使用gpu

Question

我正在尝试创建和训练一个CNN模型。但每次我运行代码时，tensorflow并没有使用GPU，而是使用CPU。我已经安装了tensorflow的最新版本。附上以下详细信息。

python => 3.9.5\
Tensorflow-GPU => 2.5.0\
CUDA => 11.3\
cuDNN => 8.2.1

在运行时，我得到以下带有警告消息的输出。（平台：VS代码）

2021-07-28 15:35:13.163991： W tenstorflow/core/common_runtime/bfc_allocator.cc:337]垃圾收集：解除分配空闲内存区域（即分配），这样我们就可以重新分配一个更大的区域，以避免内存碎片导致的OOM。如果您经常看到此消息，则表示您正在接近可用设备内存的阈值，重新分配可能会产生很大的性能开销。您可以尝试较小的批量大小来观察性能影响。如果要禁用此功能，请设置TF_ENABLE_GPU_GARBAGE_COLLECTION=false。

系统性能

输出

代码在这里可以访问

注意：我已经尝试添加以下代码来激活gpu，但它不起作用。

gpus = tf.config.experimental.list_physical_devices('GPU')
tf.config.experimental.set_memory_growth(gpus[0], True)
tf.config.set_visible_devices(gpus[0], 'GPU')

供建议的产出：

Answer 1

试试这个：

physical_devices = tf.config.list_physical_devices('GPU')
print("Num GPUs:", len(physical_devices))

由于您的型号非常小，GPU可能会受到系统其他部分的限制，无法使用完整的GPU。

另一方面，任务管理器不是检查GPU负载的最佳工具，如果可用，请使用nvidia smi。