Deep Learning Containers are a set of Docker containers with key data science frameworks, libraries, and tools pre-installed. These containers provide you with performance-optimized, consistent environments that can help you prototype and implement workflows quickly.
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
- fairness-indicators for TensorFlow 2.3 and 2.4 Deep Learning Containers instances
- many others
Nvidia packages with the latest Nvidia driver for GPU-enabled instances:
- CUDA 10.*, 11.*, and 12.* (the version depends on the framework)
- CuDNN 7.* and NCCL 2.* (the version depends on the CUDA version)
JupyterLab
Hugging Face frameworks and libraries:
- Text Generation Inference toolkit
- Text Embeddings Inference toolkit
- Transformers library
- Sentence Transformers library
- Diffusers library
Model Garden containers
- vLLM library
Community support
Ask a question about Deep Learning Containers on Stack Overflow or join the google-dl-platform Google group to discuss Deep Learning Containers.
Learn more about getting support from the community.
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.