xTag

概要

フル名
docker.io/hiyouga/pytorch
Registry
docker.io
名前空間
hiyouga
リポジトリ
pytorch
キャッシュ Tag 数
2
最終同期
07/13/2026, 03:54 PM

概要

PyTorch 深度学习基础镜像（hiyouga 构建），GPU/CUDA 预装 PyTorch 环境，LLM 微调与 ML 实验 Dockerfile 基础层。

詳細

hiyouga/pytorch 提供预装 PyTorch 与 CUDA 的深度学习运行环境，常用于 LLM 微调（LLaMA-Factory 等）与 ML 实验。定位为 PyTorch Dockerfile FROM 基础镜像，需 GPU 加速；大模型微调训练、深度学习实验或 CI GPU 训练 Job 基础层。

快速启动

docker run --rm --gpus all docker.io/hiyouga/pytorch:latest python -c "import torch; print(torch.__version__)"

推荐实践

# Dockerfile: FROM hiyouga/pytorch:2.1
# 微调训练 Job
docker run --rm --gpus all -v $(pwd):/workspace \
  docker.io/hiyouga/pytorch:2.1.0 \
  python train.py

核心参数说明

  • --gpus all — NVIDIA GPU 加速
  • tag — PyTorch/CUDA 版本
  • 用法 — 训练 Job 或 FROM 基础
  • 勿 — 长期无任务 docker run

最新 Tag（20 件）

すべての 2 件の Tag を見る →
Tagアーキテクチャプッシュ日時サイズDigest同期期望操作
docker.io/hiyouga/pytorch
:th2.6.0-cu124-flashattn2.7.4-cxx11abi0-devel
linux/amd64
2025/05/28 21:328.3 GBsha256:bb606e5cb5a1…
docker.io/hiyouga/pytorch
:th2.6.0-cu124-flashattn2.7.4-cxx11abi0
linux/amd64
2025/05/27 18:014.6 GBsha256:9e4e3deb7aa6…