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

  • Hugging Face 框架和库：

    • 文本生成推理工具包
    • 文本嵌入推理工具包
    • Transformer 库
    • 句子转换器库
    • Diffusers 库

  • Model Garden 容器

    • vLLM 库

社区支持

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

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

后续步骤

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