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 上更新