Dockerfile

FROM jupyter/base-notebook:latest

USER root
RUN pip install torch transformers accelerate bitsandbytes peft trl httpx wandb matplotlib
USER jovyan

EXPOSE 7860
CMD ["jupyter", "lab", "--ip=0.0.0.0", "--port=7860", "--no-browser", "--allow-root", "--NotebookApp.token=''"]