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…