1

我可以知道谷歌深度学习 VM docker 映像是否可以下载以在具有 Nvidia GPU 的开发人员机器上运行？它可以在其他云而不是 Google Compute Engine 上运行吗？有谁知道图像注册表的链接是什么？

4

1 回答 1

2

我可以知道谷歌深度学习 VM docker 映像是否可以下载以在具有 Nvidia GPU 的开发人员机器上运行？

我们最近创建了与 Google 深度学习 VM 完全一致的环境。每当无法使用虚拟机时都应该使用这些环境（本地开发就是一个很好的例子）。它们目前处于 Beta 阶段，许多事情仍需要完善，但很高兴能得到您的早期反馈。容器列表：

  • gcr.io/deeplearning-platform-release/base-cpu:最新
  • gcr.io/deeplearning-platform-release/base-cu100：最新
  • gcr.io/deeplearning-platform-release/pytorch-cpu.1-0:latest（Jupyter UI 在这个上还没有功能）
  • gcr.io/deeplearning-platform-release/pytorch-gpu.1-0:latest（Jupyter UI 在这个上还没有功能）
  • gcr.io/deeplearning-platform-release/tf-cpu.1-13:latest
  • gcr.io/deeplearning-platform-release/tf-gpu.1-13:latest

（请记住，将来即使命名也可能会改变）

您可以列出所有可用的容器：

gcloud container images list --repository="gcr.io/deeplearning-platform-release"

以下是在本地启动的方法：

CPU 环境

IMAGE_NAME="gcr.io/deeplearning-platform-release/base.cpu:m23"

docker pull "${IMAGE_NAME}"
docker run -p 127.0.0.1:8080:8080/tcp -v "${HOME}:/home" "${IMAGE_NAME}"

GPU 环境

  • 安装环境所需的Nvidia驱动程序（TODO）
  • 安装 Nvidia-docker
IMAGE_NAME="gcr.io/deeplearning-platform-release/base.cu100:m23"

docker pull "${IMAGE_NAME}"
docker run --runtime=nvidia -p 127.0.0.1:8080:8080/tcp -v "${HOME}:/home" "${IMAGE_NAME}"

它可以在其他云而不是 Google Compute Engine 上运行吗？

虚拟机映像 - 没有。理论上，容器是可以的，但我们不能保证它会起作用。

有谁知道图像注册表的链接是什么？

对于 VM 映像，没有映像注册表之类的概念，但是，对于容器，您可以使用我在答案开头粘贴的链接。

于 2019-04-24T18:39:32.997 回答