yo

docker/Dockerfile

@@ -266,6 +266,43 @@ RUN if [ -n "$CUDA_VERSION" ] ; then \
         cd .. ; \
     fi
 
+######################################
+# Custom dependencies for this branch
+######################################
+
+# Install system dependencies
+RUN apt-get update -qqq && \
+    apt-get install -qqq ffmpeg libsm6 libxext6 -y && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+# Install Python dependencies
+RUN pip install -qqq --no-cache-dir \
+    awscli \
+    packaging \
+    ninja \
+    timm \
+    ftfy \
+    flash-attn \
+    accelerate \
+    wandb \
+    diffusers \
+    mmengine \
+    av \
+    opencv-python \
+    pandas \
+    fairscale \
+    colossalai \
+    rotary-embedding-torch \
+    pandarallel \
+    boto3
+
+# Install Apex
+RUN pip install -qqq --disable-pip-version-check --no-cache-dir --no-build-isolation \
+    --config-settings "--build-option=--cpp_ext" \
+    --config-settings "--build-option=--cuda_ext" \
+    git+https://github.com/NVIDIA/apex.git
+
 ###############
 # Install cmake
 ###############