Overview
- Full name
docker.io/hiyouga/pytorch- Registry
- docker.io
- Namespace
- hiyouga
- Repository
- pytorch
- Cached tags
- 2
- Last synced
- 07/13/2026, 03:54 PM
Introduction
PyTorch 深度学习基础镜像（hiyouga 构建），GPU/CUDA 预装 PyTorch 环境，LLM 微调与 ML 实验 Dockerfile 基础层。
Details
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
Latest tags (20)
View all 2 tags →| Tag | Arch | Pushed At | Size | Digest | Sync | Wish | Action |
|---|---|---|---|---|---|---|---|
docker.io/hiyouga/pytorch :th2.6.0-cu124-flashattn2.7.4-cxx11abi0-devel | linux/amd64 | 2025/05/28 21:32 | 8.3 GB | sha256:bb606e5cb5a1… | |||
docker.io/hiyouga/pytorch :th2.6.0-cu124-flashattn2.7.4-cxx11abi0 | linux/amd64 | 2025/05/27 18:01 | 4.6 GB | sha256:9e4e3deb7aa6… |