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 库
    • Sentence Transformers 库
    • 扩散器库
  • Model Garden 容器

    • vLLM 库

社区支持

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

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

后续步骤

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