From dd889086f456c0fef5a8d935ff22d3e9221785cc Mon Sep 17 00:00:00 2001
From: Jaret Burkett <jaretburkett@gmail.com>
Date: Sat, 21 Sep 2024 12:07:07 -0600
Subject: [PATCH] Updates to the docker file for jupyterlab

---
 docker/Dockerfile | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/docker/Dockerfile b/docker/Dockerfile
index 0c548507..9cc947f3 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -1,4 +1,5 @@
-FROM runpod/base:0.6.2-cuda12.1.0
+FROM runpod/base:0.6.2-cuda12.2.0
+
 LABEL authors="jaret"
 
 # Install dependencies
@@ -17,5 +18,10 @@ RUN python -m pip install -r requirements.txt
 
 RUN apt-get install -y tmux nvtop htop
 
+RUN pip install jupyterlab
+
+# symlink app to workspace
+RUN ln -s /app/ai-toolkit /workspace/ai-toolkit
+
 WORKDIR /
 CMD ["/start.sh"]
\ No newline at end of file