Dockerfile

# switch to miniconda3 base as pytorch base uses python3.7
FROM continuumio/miniconda3:23.3.1-0

# permanent dependencies, put on top to avoid re-build
RUN pip install --upgrade pip
RUN apt-get update && apt-get install -y vim tmux && \
    pip install pip-tools

COPY requirements.txt .
RUN pip install -r requirements.txt

WORKDIR /workspace

RUN chgrp -R 0 . && \
    chmod -R g=u .

RUN chgrp -R 0 /opt/conda && \
    chmod -R g=u /opt/conda

# tools
RUN apt-get update && apt-get install -y \
	vim \
	nmon

# put this at the end as we change this often, we add dummy steps to force rebuild the following lines when needed
# RUN pwd && pwd && pwd && pwd
RUN pip install --pre torch==2.0.1+cu117 torchvision --index-url https://download.pytorch.org/whl/cu117
RUN pip install torchmetrics==1.0.1
# RUN pip install --pre torch torchvision --index-url https://download.pytorch.org/whl/nightly/cu117

RUN python3 -m pip install jupyterlab
ENTRYPOINT ["jupyter", "lab", "--ip=0.0.0.0", "--allow-root"]