File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change 2
2
3
3
FROM nvidia/cuda:10.0-cudnn7-runtime-ubuntu18.04 AS armory
4
4
5
+ # Temporary fix for broken nvidia package checksum
6
+ RUN rm -f /etc/apt/sources.list.d/nvidia-ml.list
7
+
5
8
RUN apt-get -y -qq update && \
6
9
apt-get install -y wget vim build-essential git curl
7
10
@@ -21,14 +24,13 @@ RUN /opt/conda/bin/pip install --no-cache-dir \
21
24
tensorflow-datasets==3.2.0 \
22
25
jupyterlab==1.2.6 \
23
26
boto3==1.11.13 \
27
+ adversarial-robustness-toolbox==1.4.1 \
24
28
Pillow==7.1.2 \
25
29
pydub==0.24.1 \
26
30
apache-beam==2.22.0 \
27
31
dill==0.3.1.1 \
28
32
pytest==5.3.5
29
33
30
- RUN /opt/conda/bin/pip install git+https://github.com/Trusted-AI/adversarial-robustness-toolbox.git@c9f3bdeb79f2fd392c449bbcf2714cf11c485610
31
-
32
34
RUN /opt/conda/bin/conda install -c conda-forge ffmpeg==4.2.3 && \
33
35
/opt/conda/bin/conda clean --all
34
36
You can’t perform that action at this time.
0 commit comments