没有 nvidia-docker 有没有办法支持 GPU答案

【问题标题】：Is there a way for GPU support without nvidia-docker没有 nvidia-docker 有没有办法支持 GPU
【发布时间】：2019-09-21 13:23:14
【问题描述】：

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

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

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

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

【问题讨论】：

【解决方案1】：

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

之前，您必须安装 nvidia-docker，它是一个普通的 docker，为 nvidia GPU 提供了一层薄薄的抽象层。见-Nvidia Docker

【讨论】：

【解决方案2】：

您可以使用x11docker

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

x11docker --gpu imagename

【讨论】：

【解决方案3】：

您不能简单地在 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 版本的最低计算能力。

【讨论】：

【解决方案4】：

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

{
    "default-runtime": "nvidia"
}

【讨论】：