深度學習容器總覽

深度學習容器是一組 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

• Hugging Face 架構和程式庫：

  • 文字產生推論工具包
  • 文字嵌入推論工具包
  • 轉換工具程式庫
  • Sentence Transformers 程式庫
  • 擴香機程式庫

• Model Garden 容器

  • vLLM 程式庫

社群支援

在 Stack Overflow 上提出深度學習容器相關問題，或是加入 google-dl-platform Google 群組，討論深度學習容器。

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

後續步驟

您可依照使用指南中的步驟，開始使用深度學習容器。這些指南提供建立及使用深度學習容器的操作說明。