有没有办法在没有nvidia-docker的情况下支持GPU？

Question

我正在尝试在不使用nvidia-docker的情况下在容器上获得gpu支持

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

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

如果容器上没有nvidia docker，我如何使用tensorflow gpu？

Answer 1

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

早些时候，你必须安装nvidia-docker，这是一个简单的docker，为nvidia图形处理器提供了一个很薄的抽象层。见- Nvidia Docker

Answer 2

您可以使用x11docker

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

x11docker --gpu imagename

Answer 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版本的最低计算能力。