test dependencies

mlcommons · Jan 10, 2025 · e4f4d5f · e4f4d5f
1 parent 5744352
commit e4f4d5f
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 8 deletions.
diff --git a/stable_diffusion/Dockerfile b/stable_diffusion/Dockerfile
@@ -1,11 +1,15 @@
-ARG FROM_IMAGE_NAME=nvcr.io/nvidia/pytorch:22.12-py3
+ARG FROM_IMAGE_NAME=pytorch/pytorch:1.13.1-cuda11.6-cudnn8-devel
 FROM ${FROM_IMAGE_NAME}
 
 ENV DEBIAN_FRONTEND=noninteractive
 
 # apt dependencies
-RUN apt-get update
-RUN apt-get install -y ffmpeg libsm6 libxext6
+RUN echo 'debconf debconf/frontend select Noninteractive' | debconf-set-selections
+RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub
+RUN apt-get update && apt-get install -y
+RUN apt-get install -y ffmpeg libsm6 libxext6 git wget unzip \
+        build-essential \
+        libomp-dev
 
 # pip dependencies
 COPY requirements.txt .
@@ -19,4 +23,4 @@ RUN pip install  httpx==0.24.1
 # install LDM
 ADD . /diffusion
 RUN chmod +x /diffusion/*.sh
-WORKDIR /diffusion
+WORKDIR /diffusion
diff --git a/stable_diffusion/requirements.txt b/stable_diffusion/requirements.txt
@@ -20,6 +20,7 @@ colossalai==0.2.7
 invisible-watermark==0.1.5
 diffusers==0.14.0
 cloudpathlib==0.13.0
-git+https://github.com/facebookresearch/xformers.git@5eb0dbf315d14b5f7b38ac2ff3d8379beca7df9b#egg=xformers
+#git+https://github.com/facebookresearch/xformers.git@5eb0dbf315d14b5f7b38ac2ff3d8379beca7df9b#egg=xformers
+xformers==0.0.16
 bitsandbytes==0.37.2
-git+https://github.com/mlcommons/logging.git@8405a08bbfc724f8888c419461c02d55a6ac960c
+git+https://github.com/mlcommons/logging.git@8405a08bbfc724f8888c419461c02d55a6ac960c
diff --git a/stable_diffusion/scripts/datasets/mlcube_demo_data.sh b/stable_diffusion/scripts/datasets/mlcube_demo_data.sh
@@ -16,5 +16,5 @@ mkdir -p ${DEMO_OUTPUT_DIR}
 cd ${DEMO_OUTPUT_DIR}
 
 wget -O demo_data.zip -c https://mlcube.mlcommons-storage.org/minibenchmarks/stable_diffusion.zip
-unzip -o stable_diffusion.zip
-rm stable_diffusion.zip
+unzip -o demo_data.zip
+rm demo_data.zip