diff --git a/docker/Dockerfile.template b/docker/Dockerfile.template index c4f046d9352fd8b706a6bf0227b89511ce7dd6f9..9ec6a35f94e1e48fa945e26fa14b2e5313ea2b6f 100644 --- a/docker/Dockerfile.template +++ b/docker/Dockerfile.template @@ -43,9 +43,9 @@ ENV PATH "/home/opam/.local/bin:$PATH" # Python Conda -ENV SHACONDA "1564571a6a06a9999a75a6c65d63cb82911fc647e96ba5b729f904bf00c177d3" +ENV SHACONDA "aef279d6baea7f67940f16aad17ebe5f6aac97487c7c03466ff01f4819e5a651" -RUN wget https://repo.anaconda.com/miniconda/Miniconda3-py39_23.3.1-0-Linux-x86_64.sh -O ~/miniconda.sh +RUN wget https://repo.anaconda.com/miniconda/Miniconda3-py310_23.3.1-0-Linux-x86_64.sh -O ~/miniconda.sh RUN SHAFILE=$(sha256sum ~/miniconda.sh | awk '{print $1}') && if [ "$SHAFILE" != "$SHACONDA" ]; then echo "Mismatch between SHA256SUM of downloaded script and expected script, aborting installation."; echo "The expected SHA256SUM is $SHACONDA, but the SHA256SUM of downloaded script is $SHAFILE."; exit 1; fi @@ -113,6 +113,25 @@ pip3 install -r requirements.txt ENV PYTHONPATH "$PYTHONPATH:/home/opam/nnenum/src" +## ABCrown + +RUN git clone https://github.com/Verified-Intelligence/alpha-beta-CROWN.git && \ +git -C alpha-beta-CROWN checkout 7b3d507caf40a5f42d99f6894373526c6fd615b7 + +### From +# https://github.com/Verified-Intelligence/alpha-beta-CROWN/blob/main/vnncomp_scripts/install_tool_general.sh, +# and removing dependencies that are already in PyRAT +# dependency set: pytorch, torchvision, torchaudio, tqdm, +# pandas +# DNNV and onnx2pytorch set on a specific commit +# WARNING: Result in Pillow version and networkx version +# different than what is expected by PyRAT and/or nnenum! + +RUN cd alpha-beta-CROWN && \ +conda install --yes cudatoolkit=11.1 mkl=2020.0 pyyaml packaging appdirs protobuf sortedcontainers -c pytorch-lts -c nvidia -c gurobi && \ +pip install --no-input --no-cache-dir onnx onnxruntime git+https://github.com/KaidiXu/onnx2pytorch.git@fe7281b9b6c8c28f61e72b8f3b0e3181067c7399 git+https://github.com/dlshriver/DNNV.git@6ab81fff8b83e5e7d5d964e409d092190368c8f1 + +ENV PYTHONPATH "$PYTHONPATH:/home/opam/alpha-beta-CROWN/complete_verifier" # CAISAR