深度學習容器總覽

Deep Learning Containers 是一組 Docker 容器，已預先安裝重要的資料科學架構、程式庫和工具。這些容器提供效能最佳化的一致環境，可協助您快速設計原型及部署工作流程。

如要進一步瞭解容器，請參閱「Google 容器」。

預先安裝軟體

您可以設定深度學習容器映像檔，加入下列項目：

• 架構：

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

• Python，包括下列套件：

  • numpy
  • sklearn
  • scipy
  • pandas
  • nltk
  • pillow
  • 適用於 TensorFlow 2.3 和 2.4 深度學習容器執行個體的 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。

進一步瞭解如何取得社群支援。

後續步驟

如要開始使用深度學習容器，請參閱使用指南，瞭解如何建立及使用深度學習容器。