Skip to content

Commit

Permalink
installing sscd requirements directly in Dockerfile
Browse files Browse the repository at this point in the history
  • Loading branch information
ahmednasserswe committed Nov 8, 2023
1 parent 3c3fc9f commit 8224cfb
Showing 1 changed file with 14 additions and 3 deletions.
17 changes: 14 additions & 3 deletions Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 8224cfb

Please sign in to comment.