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

Question

大家好，谢谢。

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

我从https://github.com/NVIDIA/nvidia-docker/wiki/Installation-(version-2.0)开始检查了前提条件。

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

容器定期启动，我可以使用笔记本，但不支持GPU ...

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

仅可识别CPU ...

我可能缺少明显的东西...我是docker和tensorflow的新手...

感谢您的帮助！

Answer 1

对不起，缺少--runtime = nvidia ...

docker run --runtime = nvidia -u $ {id -u}：$ {id -g）-it --rm -v $ {realpath〜/ Projects / GDL / GDL_code）：/ tf / notebooks -p 8888：8888张量/张量：2.0.0-gpu-py3-jupyter

还是谢谢大家！