google-cloud 文档
Google Cloud 上可用的 DLC
加入 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 |