Dockerでjupyter lab

FROM ubuntu:24.04

RUN apt update -y
RUN apt upgrade -y
RUN apt install -y python3-venv

# rootユーザーでの実行はセキュリティリスクがあるのでユーザーを作成する
RUN useradd -m -s /bin/bash lab

# 実行ユーザーを変更
USER lab
WORKDIR /home/lab/

# ubuntuでのPython実行は仮想環境が推奨されるので作成
RUN python3 -m venv /home/lab/venv

# /bin/shで実行されるがshにはsourceコマンドがないためbashを指定する
RUN bash -c "source /home/lab/venv/bin/activate && pip install jupyterlab opencv-python"

CMD ["bash", "-c", "source /home/lab/venv/bin/activate && jupyter-lab --ip=0.0.0.0 --no-browser --port=8888 --NotebookApp.token=''"]

FROM nvidia/cuda:12.8.1-runtime-ubuntu24.04

RUN apt update -y
RUN apt upgrade -y
RUN apt install -y python3-venv python3-opencv
RUN apt install -y git nodejs

# rootユーザーでの実行はセキュリティリスクがあるのでユーザーを作成する
RUN useradd -m -s /bin/bash torch

# 実行ユーザーを変更
USER torch
WORKDIR /home/torch/

# ubuntuでのPython実行は仮想環境が推奨されるので作成
RUN python3 -m venv /home/torch/venv

# /bin/shで実行されるがshにはsourceコマンドがないためbashを指定する
RUN bash -c "source /home/torch/venv/bin/activate && pip install --upgrade pip"
RUN bash -c "source /home/torch/venv/bin/activate && pip install torch torchvision"
RUN bash -c "source /home/torch/venv/bin/activate && pip install jupyterlab opencv-python"
# 必要に応じて
RUN bash -c "source /home/torch/venv/bin/activate && pip install pandas scikit-learn matplotlib"
RUN bash -c "source /home/torch/venv/bin/activate && pip install tqdm albumentations"

RUN bash -c "source /home/torch/venv/bin/activate && pip install ipywidgets jupyterlab_widgets"


CMD ["bash", "-c", "source /home/torch/venv/bin/activate && jupyter-lab --ip=0.0.0.0 --no-browser --port=8888 --NotebookApp.token=''"]