tensorflow:2.0.0-docker容器无法访问GPU

大家好，提前谢谢。

我的目标是使用一个支持GPU的容器来执行机器学习手册（第二版）的笔记本。
我的想法是使用支持GPU的容器，可能添加一些导入，然后承诺创建一个新映像

我从中检查了先决条件。

O.S.Ubuntu 18.04
处理器：英特尔®核心™ i7-7700HQ CPU@2.80GHz×8
图形卡：GeForce GTX 1080/PCIe/SSE2
NVIDIA-SMI 418.87.00，图形卡可识别，
docker版本：19.03.5 API版本：1.40，
nvidia-docker2已经是最新版本（2.2.2-1），
nvidia docker（旧版本）不存在

正在执行：

docker pull tensorflow/tensorflow:2.0.0-gpu-py3-jupyter<br/><br/>
docker run -u $(id -u):$(id -g) -it --rm -v $(realpath ~/Projects/GDL/GDL_code):/tf/notebooks -p 8888:8888 tensorflow/tensorflow:2.0.0-gpu-py3-jupyter

只有CPU被识别

我可能错过了一些明显的。。。我是docker和tensorflow的新手…

感谢您的帮助

很抱歉，缺少--runtime=nvidia

docker run--runtime=nvidia-u$（id-u）：$（id-g）-it--rm-v$（realpath~/Projects/GDL/GDL_-code）：/tf/notebooks-p888:8888 tensorflow/tensorflow:2.0.0-gpu-py3-jupyter

---

谢谢大家

哎呀。。。缺少--runtime=nvidia…docker run--runtime=nvidia-u$（id-u）：$（id-g）-it--rm-v$（realpath~/Projects/GDL/GDL\u code）：/tf/notebooks-p 8888:8888 tensorflow/tensorflow:2.0.0-gpu-py3-jupyteri如果通过设置

--runtime

标志解决了问题，请将其作为答案写在下面，并将其标记为解决方案。如果您的问题仍未解决，请参考此答案-

import tensorflow as tf
from tensorflow.python.client import device_lib