Initial commit

FROM tensorflow/tensorflow:nightly-gpu

# https://jupyterhub.readthedocs.io/en/stable/tutorial/quickstart.html

# https://metamath1.github.io/2018/12/24/jupyterhub.html

RUN apt-get update -qq \

   && apt-get install -yqq --no-install-recommends \

      nodejs \

      npm \

      sudo \

      wget \

      git \

      nano \

      vim \

   && rm -rf /var/lib/apt/lists/*

RUN npm install -g configurable-http-proxy

RUN python3 -m pip install --no-cache-dir jupyterhub jupyterlab sudospawner tensorflow

RUN python3 -m pip install --no-cache-dir jupyterlab-git pandas seaborn beautifulsoup4 bokeh h5py ipympl scipy ipywidgets

RUN groupadd jupyterhub

RUN useradd -rm -d /home/rhea -s /bin/bash -G sudo -u 1000 -g jupyterhub rhea

RUN usermod -aG shadow rhea

RUN mkdir /etc/jupyterhub

RUN chown rhea /etc/jupyterhub

WORKDIR /etc/jupyterhub

# USER rhea

RUN sudo -u rhea jupyterhub --generate-config

CMD ["sudo", "-u", "rhea", "jupyterhub", "--JupyterHub.spawner_class=sudospawner.SudoSpawner"]

version: '3.3'

services:

  jupyter-gpu:

    container_name: jupyterhub-gpu

    build: .

    ports:

      - '28000:8000'

    volumes:

      - ${PWD}/config:/etc/jupyterhub

      - ${PWD}/home:/home

      - ${PWD}/share:/share

      - ./passwd/sudoers:/etc/sudoers

    image: 'jupyterhub-test'

    restart: always

    #devices:

    #  - /dev/nvidia0:/dev/nvidia0

    #  - /dev/nvidiactl:/dev/nvidiactl

    #  - /dev/nvidia-caps:/dev/nvidia-caps

    #  - /dev/nvidia-modeset:/dev/nvidia-modeset

    #  - /dev/nvidia-uvm:/dev/nvidia-uvm

    #  - /dev/nvidia-uvm-tools:/dev/nvidia-uvm-tools

    #environment:

    # - NVIDIA_VISIBLE_DEVICES=0

    # - CUDA_VISIBLE_DEVICES=0

    # - NVIDIA_VISIBLE_DEVICES: "ALL" 

    deploy:

      resources:

        reservations:

          devices:

            - driver: nvidia

              # count: all

              device_ids: ["0"]

              capabilities: [gpu]