Overview
- Full name
docker.io/cnstark/pytorch- Registry
- docker.io
- Namespace
- cnstark
- Repository
- pytorch
- Cached tags
- 119
- Last synced
- 07/05/2026, 10:16 PM
Introduction
PyTorch 深度学习基础镜像，预装 CUDA/PyTorch 框架，ML 训练与推理 Dockerfile 基础层。
Details
cnstark/pytorch 提供预装 PyTorch 与 CUDA 的深度学习基础环境，作为 ML 训练/推理 Dockerfile 的 FROM 基础层或 Jupyter 开发容器。定位为基础 ML 框架镜像，本身无业务入口，需 --gpus all 挂载 GPU；非独立 Web 服务，无常规 K8s Deployment 场景。
快速启动
docker run --rm docker.io/cnstark/pytorch:2.1.0-cuda12.1 \
python -c "import torch; print(torch.__version__)"
推荐实践
# Dockerfile 基础层
# FROM docker.io/cnstark/pytorch:2.1.0-cuda12.1
# COPY train.py /app/train.py
# CMD ["python", "/app/train.py"]
核心参数说明
python -c import torch— 验证 PyTorch- 用途 — Dockerfile FROM 基础层
- GPU — 需
--gpus all - 勿 — 裸镜像独立 K8s 部署