Update Dockerfile

Author: daniel-unyi-42

Dockerfile

@@ -1,27 +1,37 @@
+# Base image with CUDA 12.1 runtime and cuDNN
 FROM nvidia/cuda:12.1.1-cudnn8-runtime-ubuntu22.04
 
+# Install essential system tools and Python
 RUN apt-get update -y && apt-get install -y --no-install-recommends \
     git \
     wget \
-    tmux \
-    vim \
-    htop \
-    zip \
+    curl \
     unzip \
+    htop \
+    vim \
     build-essential \
     python3 \
-    python3-pip && \
-    apt-get clean && \
+    python3-pip \
+    python3-venv \
+    && apt-get clean && \
     rm -rf /var/lib/apt/lists/*
 
-RUN python3 -m pip install --no-cache-dir --upgrade pip && \
-    pip install --no-cache-dir debugpy
+# Create a symbolic link to enable `python` instead of `python3`
+RUN ln -s /usr/bin/python3 /usr/bin/python
 
+# Install global python tools
+RUN python -m pip install --no-cache-dir --upgrade pip && \
+    pip install --no-cache-dir debugpy jupyterlab
+
+# Set working directory
 WORKDIR /workspace
 
+# Clone the repository and install segger with CUDA 12 support
 RUN git clone https://github.com/EliHei2/segger_dev.git /workspace/segger_dev && \
-    pip install -e "/workspace/segger_dev[cuda12,rapids12,cupy12,faiss]"
-
-EXPOSE 5678
+    pip install -e "/workspace/segger_dev[cuda12]"
 
+# Set environment variables
 ENV PYTHONPATH=/workspace/segger_dev/src:$PYTHONPATH
+
+# expose ports for debugpy and jupyterlab
+EXPOSE 5678 8888