在 Docker 中为 Pytorch 模型访问 GPU

【问题标题】：Accessing GPU in Docker for Pytorch Model在 Docker 中为 Pytorch 模型访问 GPU
【发布时间】：2021-10-05 02:06:20
【问题描述】：

我开发了一个机器学习模型并将其与 Flask 应用程序集成。当我尝试为该应用程序运行 docker 映像时，它说我没有 GPU 访问权限。我应该如何编写一个 Dockerfile 以便我可以在容器内使用“cuda gpu”？以下是 Dockerfile 的当前状态。

来自 python:3.9

WORKDIR /myapp

添加。 /我的应用程序

运行 pip3 install -r requirements.txt

复制。 .

CMD [ "python","./app.py" ]

【问题讨论】：

【解决方案1】：

执行docker run时需要使用--gpus参数，查看documentation。

【讨论】：

嗨，我应该按如下方式运行它吗？ docker container run --gpus all myapp
是的，这是一种选择 - 但我建议尝试一下。
我正在阅读文档；但是还有一点我没有得到。它仅使用 apt-get 在文档中安装 nvidia 容器运行时。在dockerfile中不使用“FROM ubuntu”就不能使用cuda gpu吗？
这是一个完全独立的问题，但我认为您确实需要定义一些操作系统。