uodcvip/pytorch

PyTorch GPU Jupyter镜像文档

镜像概述

本镜像为基于PyTorch的GPU镜像重建版本，主要用于运行dockerspawner的JupyterHub服务器环境，提供JupyterLab IDE或Jupyter notebook单用户服务器功能。构建仓库位于https://github.com/UoD-CVIP/ServerImageBuilder/%E3%80%82

核心功能与特性

• 版本兼容性：严格匹配以下Jupyter包版本以确保JupyterHub实例稳定运行：

  jupyterhub>=1.1.0<=1.2.0
  jupyterlab>=1.2.1<=1.3.0
  

• 扩展支持：包含jupyter-nvdashboard等扩展组件，增强Jupyter环境功能。

使用场景与适用范围

适用于需要在JupyterHub环境下通过dockerspawner部署单用户Jupyter服务器的场景，特别适合需要GPU加速的PyTorch相关开发和运行任务。

使用方法与配置说明

前置要求

• 安装nvidia-container-runtime以支持GPU资源调用。

JupyterHub配置

需在jupyterhub_config.py中添加以下配置：

1. 额外主机配置

python
c.Spawner.extra_host_config = {
    'runtime': 'nvidia',
    'shm_size': '16G',  # 默认值过小，需显式设置
}

2. 目录挂载配置

（注：以下挂载目录将在未来版本中移除）

python
c.Spawner.volumes = {
    'tensorboard': '/shares/local/tensorboard',
    'cvip': '/shares/local/cvip',
    '/mnt/share': '/shares/share',
    '/mnt/studentshare': '/shares/studentshare',
}

确认配置后，JupyterHub将使用本镜像为用户生成具备GPU支持的JupyterLab或Jupyter notebook环境。