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

RunPod PyTorch便捷镜像

镜像概述

本镜像为RunPod平台设计，基于官方PyTorch镜像（[***]

核心功能与特性

• 基于官方PyTorch镜像：继承PyTorch官方镜像的完整功能，适用于深度学习模型开发与训练
• 灵活服务配置：通过环境变量一键启用关键服务：
  • SSH服务：配置公钥后自动在22端口启动SSH守护进程，支持远程登录
  • Jupyter Lab服务：预配置密码后自动在8888端口启动Jupyter Lab，支持交互式开发

使用场景

• RunPod平台上的PyTorch模型开发与训练
• 需要远程访问容器环境的场景（通过SSH）
• 需要交互式编程环境的场景（通过Jupyter Lab）

环境变量配置

使用方法

基本部署（docker run）

示例1：启动SSH服务

bash
docker run -e PUBLIC_KEY="ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAABAQ..." runpod/pytorch

• 替换PUBLIC_KEY值为实际的SSH公钥
• SSH连接：ssh root@<容器IP> -p 22

示例2：启动Jupyter Lab服务

bash
docker run -p 8888:8888 -e JUPYTER_PASSWORD="your_secure_password" runpod/pytorch

• 访问Jupyter Lab：浏览器打开http://<容器IP>:8888，输入密码your_secure_password

示例3：同时启动SSH和Jupyter Lab

bash
docker run -p 22:22 -p 8888:8888 \
  -e PUBLIC_KEY="ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAABAQ..." \
  -e JUPYTER_PASSWORD="your_secure_password" \
  runpod/pytorch

注意事项

• 确保容器端口映射正确（如-p 22:22、-p 8888:8888）以暴露服务
• 公钥和密码需妥善保管，避免未授权访问
• 基于PyTorch官方镜像，可直接使用torch等深度学习库进行模型开发