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.
Deep Learning Containers images can be configured to include the following:
Python, including these packages:
- 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)
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.