Question

我正在尝试在容器上获得gpu支持没有nvidia-docker

我知道使用nvidia泊坞窗，我只需要使用 --runtime = nvidia，但我目前的情况不允许使用nvidia-docker

我尝试在容器上安装nvidia驱动程序cuda，cudnn，但失败。

如何在容器上没有nvidia docker的情况下使用tensorflow gpu？

Answer 1

您可以使用x11docker

使用gpu在X上运行docker镜像非常简单

x11docker --gpu imagename

Answer 2

您不能简单地在Docker容器中安装nvidia驱动程序。容器必须有权访问硬件。虽然我不确定，但是安装可能会帮助您解决该问题。请参阅-https://docs.docker.com/storage/

您可以使用anaconda 来安装和使用Tensorflow-gpu。

确保已安装最新的nvidia驱动程序。从官方站点安装Anaconda 2或3。 https://www.anaconda.com/distribution/

创建一个新环境并安装tensorflow-gpu和cudatoolkit。

$conda create -n tf-gpu tensorflow-gpu python cudnn cudatoolkit

您还可以指定应用程序的版本。

例如$conda create -n tf-gpu tensorflow-gpu python=3.5 cudnn cudatoolkit=8

请检查您的硬件是否具有最低的计算能力，以支持您正在/将要使用的CUDA版本。

Answer 3

如果您不能通过--runtime=nvidia作为命令行选项（例如docker-compose），则可以在Docker守护程序配置文件/etc/docker/daemon.json中设置默认运行时：

{
    "default-runtime": "nvidia"
}

Answer 4

您会很高兴知道最新的Docker版本现在支持nvidia GPU。您需要使用--device标志来指定您的Nvidia驱动程序。请参阅-How to use GPU a docker container

更早之前，您必须安装nvidia-docker，它是普通的docker，带有用于nvidia GPU的抽象薄层。请参阅-Nvidia Docker