Google Cloud 文档

Google Cloud 上可用的深度学习容器

Hugging Face's logo
加入 Hugging Face 社区

并获得增强型文档体验

开始

Google Cloud 上可用的 DLC

您可以在下面找到 Google Cloud 上所有可用深度学习容器 (DLC) 的列表。

以下列表仅包含每个 Hugging Face DLC 的最新版本,可以在 Google Cloud 深度学习容器文档Google Cloud Artifact Registry 或通过 gcloud container images list --repository="us-docker.pkg.dev/deeplearning-platform-release/gcr.io" | grep "huggingface-" 命令中找到 Google Cloud 中所有可用已发布容器的完整列表。

文本生成推理 (TGI)

容器 URI 路径 加速器
us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-generation-inference-cu124.2-3.ubuntu2204.py311 text-generation-inference-gpu.2.3.1 GPU

文本嵌入推理 (TEI)

容器 URI 路径 加速器
us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-embeddings-inference-cu122.1-4.ubuntu2204 text-embeddings-inference-gpu.1.4.0 GPU
us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-text-embeddings-inference-cpu.1-4 text-embeddings-inference-cpu.1.4.0 CPU

PyTorch 推理

容器 URI 路径 加速器
us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-pytorch-inference-cu121.2-2.transformers.4-44.ubuntu2204.py311 huggingface-pytorch-inference-gpu.2.2.2.transformers.4.44.0.py311 GPU
us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-pytorch-inference-cpu.2-2.transformers.4-44.ubuntu2204.py311 huggingface-pytorch-inference-cpu.2.2.2.transformers.4.44.0.py311 CPU

PyTorch 训练

容器 URI 路径 加速器
us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-pytorch-training-cu121.2-3.transformers.4-42.ubuntu2204.py310 huggingface-pytorch-training-gpu.2.3.0.transformers.4.42.3.py310 GPU
< > 在 GitHub 上更新