morpheus9631/jupyter

镜像概述

本Docker镜像为AI算法学习与开发提供一站式便捷环境，整合了SSH远程管理服务、CUDA/CuDNN GPU加速组件、Miniconda包管理工具，以及TensorFlow-GPU、PyTorch深度学习框架和Jupyter Notebook交互式开发平台，帮助用户快速搭建AI开发环境。

核心功能

• 远程管理：集成SSH服务，支持远程连接容器进行配置和管理
• GPU加速：预装CUDA和CuDNN，可利用GPU提升模型训练效率
• 多框架支持：包含TensorFlow-GPU和PyTorch两种主流深度学习框架
• 交互式开发：内置Jupyter Notebook，提供可视化代码编写与运行环境
• 数据持久化：支持本地目录挂载，确保代码和数据不丢失

使用场景

• AI算法入门学习与实践
• 深度学习模型开发与调试
• 远程协作进行AI项目开发
• 利用GPU加速完成大规模模型训练

配置说明

部署示例

TensorFlow-GPU版本

bash
docker run --rm -d \
 --gpus all \
 -p 2022:22 \
 -p 8888:8888 \
 -v $(pwd)/work:/home/user/work \
 --name jupyter-tfgpu \
 docker.xuanyuan.run/morpheus9631/jupyter:1.0-tfgpu-cuda10.1-cudnn7-devel

PyTorch版本

bash
docker run --rm -d \
 --gpus all \
 -p 2022:22 \
 -p 8888:8888 \
 -v $(pwd)/work:/home/user/work \
 --name jupyter-torch \
 docker.xuanyuan.run/morpheus9631/jupyter:1.0-torch-cuda10.1-cudnn7-devel

同时支持TensorFlow-GPU和PyTorch版本

bash
docker run --rm -d \
 --gpus all \
 -p 2022:22 \
 -p 8888:8888 \
 -v $(pwd)/work:/home/user/work \
 --name jupyter-torch-tfgpu \
 docker.xuanyuan.run/morpheus9631/jupyter:1.1-torch-tfgpu-cuda10.1-cudnn7-devel

环境测试

验证TensorFlow版本

bash
python -c 'import tensorflow as tf; print(tf.__version__)'

验证PyTorch版本

bash
python -c 'import torch; print(torch.__version__)'

验证GPU支持（基于TensorFlow）

bash
python -c 'from tensorflow.python.client import device_lib; print(device_lib.list_local_devices())'

远程连接

SSH连接容器

bash
ssh user@[容器IP] -p 2022
# 密码：1234

Jupyter操作

启动Jupyter Notebook

bash
start-notebook.sh
# 将输出中的"[容器IP]:8888/?token=xxxxxx"复制到浏览器访问

停止Jupyter Notebook

bash
stop-notebook.sh