batchlogit/Dockerfile at main · frankier/batchlogit · 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
FROM debian:bullseye-slim

LABEL org.opencontainers.image.source https://github.com/frankier/batchlogit

RUN apt-get update && \
    DEBIAN_FRONTEND=noninteractive \
    apt-get dist-upgrade -yq && \
    DEBIAN_FRONTEND=noninteractive \
    apt-get install -yq \
    git \
    build-essential \
    wget \
    bash \
    coreutils \
    ca-certificates \
    curl \
    && \
    rm -rf /var/lib/apt/lists/*

RUN curl -L https://micromamba.snakepit.net/api/micromamba/linux-64/0.7.14 | \
    tar -xj -C / bin/micromamba

ENV LANG=C.UTF-8 LC_ALL=C.UTF-8
ENV MAMBA_ROOT_PREFIX=/opt/conda
ENV PATH "$MAMBA_ROOT_PREFIX/bin:$PATH"
RUN mkdir $MAMBA_ROOT_PREFIX && \
    yes | /bin/micromamba shell init -s bash -p $MAMBA_ROOT_PREFIX

WORKDIR /opt/batchlogit

RUN micromamba install -y \
       -n base \
       -c nvidia \
       -c conda-forge \
       python=3.8.8 \
       'cudatoolkit=10.2'

COPY environment.yml ./

RUN micromamba install -y \
       -n base \
       -c rapidsai-nightly \
       -c nvidia \
       -c conda-forge \
       -c defaults \
       -c anaconda \
       -f ./environment.yml \
       && \
    rm /opt/conda/pkgs/cache/*

COPY . /opt/batchlogit

RUN echo "/opt/batchlogit" > \
    /opt/conda/lib/python3.8/site-packages/batchlogit.pth