puzlcloud/pytorch Docker 镜像 - 轩辕镜像 | Docker 镜像高效稳定拉取服务

PyTorch Docker镜像

镜像概述和主要用途

PyTorch机器学习框架，包含多种Python运行时环境。该镜像为非root用户设计，由Puzl Kubernetes云服务使用，基于官方CUDA Docker镜像构建。

支持的语言和解释器版本

• python3.10
• python3.11

已安装的包

操作系统层面

• rclone
• conda
• openssh-server
• rsync
• CUDA，版本11.8

Python层面

• torch，版本2.1.0
• torchvision，版本0.16.0
• torchaudio，版本2.1.0
• jupyterlab，版本4.0.7

使用场景和适用范围

适用于需要PyTorch框架进行机器学习模型开发、训练和推理的场景，尤其适合在Kubernetes环境中部署的AI应用，支持CUDA加速以提升计算性能。

使用方法说明

基本运行命令

bash
docker run --gpus all -it --rm pytorch-image:latest python --version

启动JupyterLab

bash
docker run --gpus all -p 8888:8888 -it --rm pytorch-image:latest jupyter lab --ip=0.0.0.0 --allow-root

注意：该镜像为非root用户设计，生产环境中建议配合适当的安全策略使用。