深度学习VM Docker注册表

时间:2019-04-24 01:58:46

标签: google-dl-platform

我可以知道Google深度学习VM泊坞映像可以下载到在具有Nvidia GPU的开发人员计算机上运行吗?可以在其他云而不是Google Compute Engine上运行吗?有人知道图像注册表的链接是什么吗?

1 个答案:

答案 0 :(得分:0)

  

我是否可以下载Google深度学习VM docker映像以在具有Nvidia GPU的开发人员计算机上运行?

我们最近创建了与Google深度学习VM完全一致的环境。每当无法使用VM时都应使用这些环境(本地开发是一个很好的示例)。它们目前处于Beta测试阶段,许多内容仍需要完善,但是很高兴获得您的早期反馈。容器清单:

  • gcr.io/deeplearning-platform-release/base-cpu:latest
  • gcr.io/deeplearning-platform-release/base-cu100:latest
  • 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

(请记住,即使将来命名也会改变)

您可以使用以下方法列出所有可用的容器:

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

这是您可以在本地开始的方式:

CPU Env

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 Env

  • 安装环境(TODO)所需的Nvidia驱动程序
  • 安装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映像-否。从理论上讲,容器是可以的,但是我们不能保证它会起作用。

  

有人知道图像注册表的链接是什么吗?

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