Skip to content

Commit 17db239

Browse files
chore: pin deps in dockerfile for future version updates (#32)
1 parent 8f5d28e commit 17db239

File tree

1 file changed

+18
-10
lines changed

1 file changed

+18
-10
lines changed

airtbench/container/Dockerfile

Lines changed: 18 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,20 @@
1-
FROM jupyter/scipy-notebook
1+
FROM jupyter/scipy-notebook@sha256:fca4bcc9cbd49d9a15e0e4df6c666adf17776c950da9fa94a4f0a045d5c4ad33
22

33
RUN pip install \
4-
torch \
5-
torchvision \
6-
torchaudio \
7-
catboost \
8-
GPy \
9-
lightgbm \
10-
xgboost \
11-
kornia \
12-
lief
4+
adversarial-robustness-toolbox==1.19.1 \
5+
catboost==1.2.8 \
6+
eagerpy==0.30.0 \
7+
foolbox==3.3.4 \
8+
GPy==1.13.2 \
9+
kornia==0.8.1 \
10+
lief==0.16.5 \
11+
lightgbm==4.6.0 \
12+
numpy==1.24.3 \
13+
plotly==6.0.1 \
14+
requests==2.31.0 \
15+
torch==2.7.0 \
16+
torchaudio==2.7.0 \
17+
torchvision==0.22.0 \
18+
xgboost==3.0.0
19+
20+
RUN pip install kornia_rs==0.1.9

0 commit comments

Comments
 (0)