1FROM nvcr.io/nvidia/cuda:11.0.3-cudnn8-devel-ubuntu20.04
2LABEL maintainer="bigscience-workshop"
3LABEL repository="petals"
4
5WORKDIR /home
6# Set en_US.UTF-8 locale by default
7RUN echo "LC_ALL=en_US.UTF-8" >> /etc/environment
8
9# Install packages
10RUN apt-get update && apt-get install -y --no-install-recommends \
11  build-essential \
12  wget \
13  git \
14  && apt-get clean autoclean && rm -rf /var/lib/apt/lists/{apt,dpkg,cache,log} /tmp/* /var/tmp/*
15
16RUN wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O install_miniconda.sh && \
17  bash install_miniconda.sh -b -p /opt/conda && rm install_miniconda.sh
18ENV PATH="/opt/conda/bin:${PATH}"
19
20RUN conda install python~=3.10.12 pip && \
21    pip install --no-cache-dir "torch>=1.12" && \
22    conda clean --all && rm -rf ~/.cache/pip
23
24VOLUME /cache
25ENV PETALS_CACHE=/cache
26
27COPY . petals/
28RUN pip install --no-cache-dir -e petals
29
30WORKDIR /home/petals/
31CMD bash
32