Deep Learning Containers 概览

Deep Learning Containers 是一组 Docker 容器，其中预安装了关键数据科学框架、库和工具。这些容器可为您提供性能更出色的一致环境，帮助您快速为工作流设计原型并加以实施。

如需详细了解容器，请参阅 在 Google 平台上使用容器。

预装软件

Deep Learning Containers 映像可以配置为包含以下内容：

• 框架：

  • TensorFlow
  • PyTorch
  • R
  • scikit-learn
  • XGBoost

• Python，包括以下软件包：

  • numpy
  • sklearn
  • scipy
  • pandas
  • nltk
  • pillow
  • TensorFlow 2.3 和 2.4 Deep Learning Containers 实例的 fairness-indicators
  • 其他

• Nvidia 软件包，其中包含适用于支持 GPU 的实例的最新 Nvidia 驱动程序：

  • CUDA 10.*、11.* 和 12.* （具体版本取决于框架）
  • CuDNN 7.* 和 NCCL 2.*（具体版本取决于 CUDA 版本）

• JupyterLab

• Model Garden 容器

  • vLLM 库

社区支持

在 Stack Overflow 上提出有关 Deep Learning Containers 的问题，或加入 google-dl-platform Google 群组讨论 Deep Learning Containers。

详细了解如何从社区获取支持。

后续步骤

您可以逐步浏览关于如何使用 Deep Learning Containers 的说明，并参阅方法指南，从而开始使用 Deep Learning Containers。