-
Notifications
You must be signed in to change notification settings - Fork 1
/
gpu.Dockerfile
72 lines (72 loc) · 1.86 KB
/
gpu.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
65
66
67
68
69
70
71
72
# NVIDIA CUDA image as a base
# We also mark this image as "deeplearning-base" so we could use it by name
FROM nvidia/cuda:11.3.0-cudnn8-runtime-ubuntu20.04 AS deeplearning-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
# Install Python
RUN add-apt-repository ppa:deadsnakes/ppa && \
apt-get update && \
apt-get install -y --no-install-recommends \
python3.10 \
python3-setuptools \
python3-pip
# Make python3 the default python version
RUN update-alternatives --install /usr/bin/python python /usr/bin/python3 1
# 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 \
# PyTorch with CUDA 11.3 support and Torchvision
torch torchvision torchaudio \
# Upgraded version of Tensorboard with more features
tensorboardX
# Here we use a base image by its name - "deeplearning-base"
# Install additional packages
FROM deeplearning-base AS other-light
RUN pip install --no-cache-dir \
# Plotting
matplotlib \
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 (non-light)
FROM other-light AS other-heavy
RUN pip install --no-cache-dir \
tensorflow-cpu \
xgboost \
catboost