From 3b41dce8b6a16aafc8f2bc28bec91534e3f6348d Mon Sep 17 00:00:00 2001 From: Charlie Meyers Date: Sun, 15 Oct 2023 02:55:00 +0200 Subject: [PATCH] update dockerfile --- Dockerfile | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/Dockerfile b/Dockerfile index 64d129ae..c5da1af0 100644 --- a/Dockerfile +++ b/Dockerfile @@ -5,7 +5,9 @@ RUN apt-get install -y python3 python3-distutils python3-pip ffmpeg libavcodec-e RUN python3 -m pip install nvidia-pyindex nvidia-cuda-runtime-cu11 RUN git clone https://github.com/simplymathematics/deckard.git WORKDIR /deckard -RUN python3 -m pip install --editable .[pytorch_image] +RUN python3 -m pip install --editable .[torchvision,torch,tensorflow] +RUN python3 -m pip install pytest RUN git clone https://github.com/Trusted-AI/adversarial-robustness-toolbox.git RUN cd adversarial-robustness-toolbox && python3 -m pip install . RUN apt install python-is-python3 +RUN pytest test