-
Notifications
You must be signed in to change notification settings - Fork 1
/
cpu.Dockerfile
64 lines (64 loc) · 1.48 KB
/
cpu.Dockerfile
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
# Python buster official image as a base
# We also mark this image as "ml-base" so we could use it by name
FROM python:3.10.9-slim-bullseye AS ml-base
WORKDIR /
# Install essential packages
ARG DEBIAN_FRONTEND=noninteractive
RUN apt-get update && apt-get install -y --no-install-recommends \
git \
build-essential \
software-properties-common \
curl \
zip \
unzip \
net-tools \
openssh-client
# Upgrade pip
RUN pip -q install pip --upgrade
# Install all basic packages
RUN pip install --no-cache-dir \
# Jupyter itself
jupyter \
# Numpy and Pandas are required a-priori
numpy pandas matplotlib
# Here we use a base image by its name - "ml-base"
# Install additional packages
FROM ml-base AS dl-base
RUN pip install --no-cache-dir \
torch \
torchvision \
torchaudio \
--extra-index-url https://download.pytorch.org/whl/cpu
FROM dl-base AS util-pkgs
RUN pip install --no-cache-dir \
# Plotting
seaborn \
plotly \
plotly_express \
# Scientific/ML packages
scipy \
scikit-learn \
lightgbm \
sktime \
imbalanced-learn \
# Image data utils
opencv-python \
Pillow \
# NLP
transformers \
spacy \
# Hyperparamerter Optimization
scikit-optimize \
hyperopt \
optuna \
# Utilities
timm \
librosa \
albumentations \
tqdm
# Install other useful packages
FROM util-pkgs AS optional
RUN pip install --no-cache-dir \
tensorflow-cpu \
xgboost \
catboost