Update Dockerfile

Dockerfile

@@ -14,9 +14,10 @@ RUN apt-get update && apt-get install -y \
     cmake \
     build-essential \
     curl \
+    python3-pip \
     && rm -rf /var/lib/apt/lists/*
 
-# Optional: Install CUDA Toolkit (only needed if compiling from source)
+# Optional: Install CUDA Toolkit for compiling custom ops
 RUN apt-get update && apt-get install -y --no-install-recommends \
     cuda-toolkit-11-8 && \
     ln -s /usr/local/cuda-11.8 /usr/local/cuda && \