This commit is contained in:
Oleg Sheynin 2024-08-03 23:00:25 -04:00
parent ac5b48db74
commit 2150bb1f70
3 changed files with 57 additions and 0 deletions

View File

@ -0,0 +1,2 @@
ROOT_DIR=/opt/jupyter_gpu/fastai
JUPYTER_PORT=7777

View File

@ -0,0 +1,31 @@
# Relies on JUPYTER_PORT being set in docker-compose.yml
# -------------------------------------------------------
FROM pytorch/pytorch:latest
# RUN useradd -m -s /bin/bash oleg
# Install packages
# --- this version of jupyter is required to work with the widgets
RUN pip install jupyterlab==3.6.7
#
RUN pip install ipywidgets
RUN pip install pandas numpy plotly
#
RUN pip install fastbook
# Install additional applications
RUN apt update
RUN apt-get install graphviz -y
# Set environment variables, etc.
#ENV MY_ENV_VAR=myvalue
# Set the working directory
WORKDIR /workspace
# Expose the port for JupyterLab
EXPOSE ${JUPYTER_PORT}
# Command to run JupyterLab
CMD ["jupyter-lab", "--ip=0.0.0.0", "--port=7777", "--no-browser", "--allow-root", "--NotebookApp.token=''"]

View File

@ -0,0 +1,24 @@
version: '3.8'
#==================================================
# Relies on the file `.env` content for varables:
# ROOT_DIR
# JUPYTER_PORT
#==================================================
services:
jupyter:
build: ${ROOT_DIR}
image: cvtt/fastai:v0.0.1
container_name: fastai_jupyter
runtime: nvidia
restart: unless-stopped
environment:
- JUPYTER_ENABLE_LAB=yes
volumes:
- ${ROOT_DIR}/notebooks:/workspace
- ${ROOT_DIR}/jupyter_settings/user-settings:/root/.jupyter/lab/user-settings
- ${ROOT_DIR}/jupyter_settings/workspaces:/root/.jupyter/lab/workspaces
- ${ROOT_DIR}/.cache/torch:/root/.cache/torch
ports:
- "${JUPYTER_PORT}:${JUPYTER_PORT}"
shm_size: "8gb"