Skip to content

Commit cd2b836

Browse files
author
ng390
authored
1 parent 0e65034 commit cd2b836

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

docker/Dockerfile

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,9 @@
22

33
FROM nvidia/cuda:10.0-cudnn7-runtime-ubuntu18.04 AS armory
44

5+
# Temporary fix for broken nvidia package checksum
6+
RUN rm -f /etc/apt/sources.list.d/nvidia-ml.list
7+
58
RUN apt-get -y -qq update && \
69
apt-get install -y wget vim build-essential git curl
710

@@ -21,14 +24,13 @@ RUN /opt/conda/bin/pip install --no-cache-dir \
2124
tensorflow-datasets==3.2.0 \
2225
jupyterlab==1.2.6 \
2326
boto3==1.11.13 \
27+
adversarial-robustness-toolbox==1.4.1 \
2428
Pillow==7.1.2 \
2529
pydub==0.24.1 \
2630
apache-beam==2.22.0 \
2731
dill==0.3.1.1 \
2832
pytest==5.3.5
2933

30-
RUN /opt/conda/bin/pip install git+https://github.com/Trusted-AI/adversarial-robustness-toolbox.git@c9f3bdeb79f2fd392c449bbcf2714cf11c485610
31-
3234
RUN /opt/conda/bin/conda install -c conda-forge ffmpeg==4.2.3 && \
3335
/opt/conda/bin/conda clean --all
3436

0 commit comments

Comments
 (0)