python - docker中的tensorflow图像是否使用GPU？

Question

我怀疑我的 docker 是否使用GPU。
我的台式机具有GeForce 2060 super和Ryzen 5 3600。
我使用tensorflow实现了CNN。
每个图像的大小为20至50 kB。
当我仅使用CPU时，每个纪元花费23秒(约95毫秒/步)。
另一方面，当我在docker中使用tensorflor-gpu-jupyter镜像时，
每个纪元花费21秒(约83毫秒/步)。
我的docker命令如下。

docker run --gpus all --rm -p 8888:8888 -v $PWD/CNN:/tf/notebooks tensorflow/tensorflow:latest-gpu-jupyter

我听说GPU加速了5到10倍，
但对我而言却没有。
我的 docker 是否使用GPU？
FYR。我检查了使用以下命令的nvidia。

docker run --gpus all --rm nvidia/cuda nvidia-smi

Answer 1

如果主机系统已正确配置(source)，则用于Docker的官方TensorFlow镜像将启用GPU。您应该使用-gpu标签拉出图像。
最简单的检查方法:使用nvtop或nvidia-smi -l 10检查主机系统中的GPU使用情况。
请注意，要启用GPU支持，您应该已经安装了NVIDIA Container Toolkit.