Os cont�ineres de aprendizado profundo s�o um conjunto de cont�ineres do Docker com os principais frameworks, bibliotecas e ferramentas de ci�ncia de dados pr�-instalados. Esses cont�ineres oferecem ambientes consistentes e otimizados para desempenho que ajudam a prototipar e implementar fluxos de trabalho rapidamente.
Para saber mais sobre cont�ineres, consulte Cont�ineres no Google.
Software pr�-instalado
As imagens do Deep Learning Containers podem ser configuradas para incluir o seguinte:
Frameworks:
- TensorFlow
- PyTorch
- R
- scikit-learn
- XGBoost
Python, incluindo estes pacotes:
- numpy
- sklearn
- scipy
- pandas
- nltk
- pillow
- indicadores de imparcialidade para inst�ncias do Deep Learning Containers do TensorFlow 2.3 e 2.4
- muitos outros
Pacotes da Nvidia com o driver mais recente dela para inst�ncias com GPU:
- CUDA 10.*, 11.* e 12.* (a vers�o depende do framework)
- CuDNN 7.* e NCCL 2.* (a vers�o depende da vers�o CUDA)
JupyterLab
Frameworks e bibliotecas do Hugging Face:
- Kit de ferramentas de infer�ncia de gera��o de texto
- Kit de ferramentas de infer�ncia de embeddings de texto
- Biblioteca Transformers
- Biblioteca Sentence Transformers
- Biblioteca de difusores
Cont�ineres do Model Garden
- Biblioteca vLLM
Suporte da comunidade
Fa�a uma pergunta sobre os cont�ineres de aprendizado profundo no Stack Overflow ou participe do grupo do Google google-dl-platform para conversar sobre esse recurso.
Saiba mais sobre como receber suporte da comunidade.
A seguir
Consulte os guias de instru��es para dar os primeiros passos com o Deep Learning Containers. Eles fornecem instru��es sobre como criar e trabalhar com cont�ineres de aprendizado profundo.