google-cloud 文档

Google Cloud 上可用的 DLC

Hugging Face's logo
加入 Hugging Face 社区

并获得增强的文档体验

开始使用

Google Cloud 上可用的 DLC

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

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

文本生成推理 (TGI)

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

文本嵌入推理 (TEI)

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

PyTorch 推理

容器 URI 路径 加速器
us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-pytorch-inference-cu121.2-3.transformers.4-46.ubuntu2204.py311 huggingface-pytorch-inference-gpu.2.3.1.transformers.4.46.1.py311 GPU
us-docker.pkg.dev/deeplearning-platform-release/gcr.io/huggingface-pytorch-inference-cpu.2-3.transformers.4-46.ubuntu2204.py311 huggingface-pytorch-inference-cpu.2.3.1.transformers.4.46.1.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 上更新