huggingface/transformers-pytorch-gpu 提供 Hugging Face transformers 与 PyTorch CUDA 预装环境，覆盖主流 NLP/CV 模型加载与训练。定位为 GPU ML 开发/推理基础镜像，需 NVIDIA GPU 与 nvidia-container-runtime；BERT/GPT 微调实验、transformers pipeline 推理或 CI GPU 测试 Job。

快速启动

docker run --rm --gpus all \
  docker.io/huggingface/transformers-pytorch-gpu:latest \
  python -c "import transformers; print(transformers.__version__)"

推荐实践

docker run -it --gpus all \
  -v /data/models:/models \
  -v $(pwd):/workspace \
  docker.io/huggingface/transformers-pytorch-gpu:4.44.0 \
  python train.py

核心参数说明

• --gpus all — NVIDIA GPU 必需
• -v .../models:... — 模型权重缓存
• CUDA — 需匹配宿主机驱动版本
• 定位 — 开发/训练 Job 非长期 API