xai-brain-decoding-benchmark/Dockerfile at master · athms/xai-brain-decoding-benchmark · GitHub

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
FROM nvidia/cuda:11.3.1-cudnn8-devel-ubuntu20.04

ARG DEBIAN_FRONTEND=noninteractive

RUN apt update
RUN apt install \
    -y \
    --no-install-recommends \
    apt-utils \
    binutils \
    build-essential \
    ca-certificates \
    curl \
    gcc \
    git \
    htop \
    less \
    libaio-dev \
    libxext6 \
    libx11-6 \
    libglib2.0-0 \
    libxrender1 \
    libxtst6 \
    libxi6 \
    locales \
    nano \
    ninja-build \
    python3-dev \
    python3-setuptools \
    python3-venv \
    python3-pip \
    screen \
    ssh \
    sudo \
    tmux \
    unzip \
    vim \
    wget

RUN python3 -m \
    pip install \
    --no-cache-dir \
    --upgrade \
    pip

RUN python3 -m \
    pip install \
    --no-cache-dir \
    torch>=1.13.1 \
    --extra-index-url https://download.pytorch.org/whl/cu113

# pip installs for which we want to specify package versions
RUN python3 -m \
    pip install \
    --no-cache-dir \
    captum>=0.5 \
    nilearn>=0.9.0 \
    ray[tune]>=1.11.0 \
    zennit>=0.4.5

# pip install where we do not care about specific packages versions
RUN python3 -m \
    pip install \
    --no-cache-dir \
    wandb \
    seaborn \
    gitdir

CMD bash