Deep Learning Containers overview

AI Platform Deep Learning Containers are a set of Docker containers with key data science frameworks, libraries, and tools pre-installed. To learn more about containers, see Containers at Google.

Pre-installed software

Deep Learning Containers images can be configured to include the following:

  • Frameworks:

    • TensorFlow
    • PyTorch
    • R
    • scikit-learn
    • XGBoost
  • Python, including these packages:

    • numpy
    • sklearn
    • scipy
    • pandas
    • nltk
    • pillow
    • many others
  • Nvidia packages with the latest Nvidia driver for GPU-enabled instances:

    • CUDA 10.* and 11.* (the version depends on the framework)
    • CuDNN 7.* and NCCL 2.* (the version depends on the CUDA version)
  • JupyterLab

What's next

You can get started with Deep Learning Containers by walking through the How-to guides, which provide instructions on create and work with deep learning containers.