Update Dockerfile

Author: N2akiti

images/scipy-ml-notebook/Dockerfile

@@ -50,52 +50,44 @@ RUN chmod 777 /etc/datahub-profile.d/*.sh /tmp/activate.sh
 
 USER jovyan
 
-# CUDA setup w/mamba
-## TODO: Investigate this command, seems to duplicate cuda packages for nvidia (pypi + conda-forge). 
-# cuda-toolkit is a skeleton package on CUDA 12, unlike CUDA <= 11
-RUN uv pip install --extra-index-url https://pypi.nvidia.com nvidia-cuda-nvcc-cu12 nvidia-nccl-cu12 cuda-python --system && \
+RUN pip3 install --no-cache-dir --upgrade uv
+
+RUN uv pip install --system \
+        --extra-index-url https://pypi.nvidia.com \
+        nvidia-cuda-nvcc-cu12 \
+        nvidia-nccl-cu12 \
+        cuda-python \
+        --index-url "" \
+        protobuf==$PROTOBUF_VERSION \
+        # OpenCV
+        opencv-contrib-python-headless \
+        opencv-python \
+        # Other Utils
+        PyQt5 \
+        pycocotools \
+        pillow \
+        scapy \
+        --index-url https://download.pytorch.org/whl/cu126 \
+        nvidia-cudnn-cu12==$CUDNN_VERSION \
+        torch==$TORCH_VERSION \
+        torchvision \
+        torchaudio \
+        --index-url "" \
+        tensorflow==$TENSORFLOW_VERSION \
+        tensorflow-datasets \
+        tensorrt==$TENSORRT_VERSION \
+        keras==$KERAS_VERSION \
+        tf-keras==$TF_KERAS_VERSION \
+        transformers \
+        datasets \
+        accelerate \
+        huggingface-hub \
+        timm \
+    && \
     fix-permissions $CONDA_DIR && \
     fix-permissions /home/$NB_USER && \
     uv cache clean
 
-# Install scipy pip packages
-## install protobuf to avoid weird base type error. seems like if we don't then it'll be installed twice.
-## https://github.com/spesmilo/electrum/issues/7825
-RUN uv pip install --no-cache-dir protobuf==$PROTOBUF_VERSION --system && \
-    fix-permissions $CONDA_DIR && \ 
-    fix-permissions /home/$NB_USER && \
-    uv cache clean
-
-## cuda-python installed to have parity with tensorflow and cudnn
-## Install pillow<7 due to dependency issue https://github.com/pytorch/vision/issues/1712
-## tensorrt installed to fix not having libnvinfer that has caused tensorflow issues.
-RUN uv pip install opencv-contrib-python-headless opencv-python --system && \
-    fix-permissions $CONDA_DIR && \ 
-    fix-permissions /home/$NB_USER && \
-    uv cache clean
-
-RUN uv pip install PyQt5 pycocotools pillow scapy --system && \
-    fix-permissions $CONDA_DIR && \
-    fix-permissions /home/$NB_USER && \
-    uv cache clean
-
-# Install CUDA/Torch/Tensorflow/Keras w/pip
-## no purge required but no-cache-dir is used. pip purge will actually break the build here!
-## Beware of potentially needing to update these if we update the drivers.
-## Check tensorrt_env_vars.sh if you have to bump tensorrt!
-
-## tf-keras is keras 2.x for higher versions of tensorflow that would normally require keras 3
-RUN uv pip install nvidia-cudnn-cu12==$CUDNN_VERSION torch==$TORCH_VERSION torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126 --system && \
-  uv pip install tensorflow==$TENSORFLOW_VERSION tensorflow-datasets tensorrt==$TENSORRT_VERSION keras==$KERAS_VERSION tf-keras==$TF_KERAS_VERSION --system && \
-  fix-permissions $CONDA_DIR && \
-  fix-permissions /home/$NB_USER && \
-  uv cache clean
-
-RUN uv pip install transformers datasets accelerate huggingface-hub timm --system && \
-  fix-permissions $CONDA_DIR && \
-  fix-permissions /home/$NB_USER && \
-  uv cache clean
-
 USER $NB_UID:$NB_GID
 ENV PATH=${PATH}:/usr/local/nvidia/bin:/opt/conda/bin