ディープ ラーニング コンテナの概要

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 インスタンス）
    • その他

  • GPU 対応インスタンス用の最新 NVIDIA ドライバを含む NVIDIA パッケージ:

    • CUDA 10.*、11.*、12.* （バージョンはフレームワークによって異なります）
    • CuDNN 7.* と NCCL 2.*（バージョンは CUDA のバージョンによって異なります）

  • JupyterLab

  • Hugging Face のフレームワークとライブラリ:

    • Text Generation Inference ツールキット
    • Text Embeddings Inference ツールキット
    • Transformer ライブラリ
    • Sentence Transformers ライブラリ
    • Diffusers ライブラリ

  • Model Garden コンテナ

    • vLLM ライブラリ

コミュニティ サポート

Stack Overflow で Deep Learning Containers について質問したり、google-dl-platform Google グループに参加して、Deep Learning Containers についてディスカッションしたりします。

コミュニティのサポートを利用する方法の詳細。

次のステップ

Deep Learning Containers を使い始めるには、入門ガイドをご覧ください。このガイドでは、ディープ ラーニング コンテナの作成と操作の手順を確認できます。