From 8224cfba766ec48299fc6300e6d0547711a9c6bc Mon Sep 17 00:00:00 2001 From: ahmednasserswe Date: Wed, 8 Nov 2023 23:03:01 +0100 Subject: [PATCH] installing sscd requirements directly in Dockerfile --- Dockerfile | 17 ++++++++++++++--- 1 file changed, 14 insertions(+), 3 deletions(-) diff --git a/Dockerfile b/Dockerfile index 8931d1a..dcde342 100644 --- a/Dockerfile +++ b/Dockerfile @@ -9,9 +9,20 @@ WORKDIR /app ENV DEBIAN_FRONTEND=noninteractive RUN git clone https://github.com/facebookresearch/sscd-copy-detection.git -RUN cd sscd-copy-detection && python -m pip install -r ./requirements.txt -RUN mkdir models_files -RUN cd sscd-copy-detection && wget https://dl.fbaipublicfiles.com/sscd-copy-detection/sscd_disc_mixup.torchscript.pt +#RUN cd sscd-copy-detection && python -m pip install -r ./requirements.txt +RUN pip install pytorch-lightning==1.6 +RUN pip install lightning-bolts==0.4.0 +RUN pip install classy_vision +RUN pip install torch +RUN pip install torchvision +RUN pip install torchmetrics +RUN pip install faiss-gpu +RUN pip install augly +RUN pip install pandas +RUN pip install numpy +RUN pip install tensorboard +#RUN mkdir models_files +#RUN cd sscd-copy-detection && wget https://dl.fbaipublicfiles.com/sscd-copy-detection/sscd_disc_mixup.torchscript.pt RUN apt-get update && apt-get install -y ffmpeg cmake swig libavcodec-dev libavformat-dev git RUN ln -s /usr/bin/ffmpeg /usr/local/bin/ffmpeg