mirror of
https://github.com/bigscience-workshop/petals
synced 2024-11-16 06:12:50 +00:00
30 lines
904 B
Docker
30 lines
904 B
Docker
|
FROM nvcr.io/nvidia/cuda:11.6.2-cudnn8-devel-ubuntu20.04
|
||
|
|
LABEL maintainer="bigscience-workshop"
|
||
|
|
LABEL repository="petals"
|
||
|
|
|
||
|
|
WORKDIR /home
|
||
|
|
# Set en_US.UTF-8 locale by default
|
||
|
|
RUN echo "LC_ALL=en_US.UTF-8" >> /etc/environment
|
||
|
|
|
||
|
|
# Install packages
|
||
|
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
||
|
|
build-essential \
|
||
|
|
wget \
|
||
|
|
git \
|
||
|
|
ed \
|
||
|
|
&& apt-get clean autoclean && rm -rf /var/lib/apt/lists/{apt,dpkg,cache,log} /tmp/* /var/tmp/*
|
||
|
|
|
||
|
|
RUN wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O install_miniconda.sh && \
|
||
|
|
bash install_miniconda.sh -b -p /opt/conda && rm install_miniconda.sh
|
||
|
|
ENV PATH="/opt/conda/bin:${PATH}"
|
||
|
|
|
||
|
|
RUN conda install python~=3.10 pip && \
|
||
|
|
pip install --no-cache-dir "torch>=1.12" torchvision torchaudio && \
|
||
|
|
conda clean --all && rm -rf ~/.cache/pip
|
||
|
|
|
||
|
|
COPY . petals/
|
||
|
|
RUN pip install -e petals[dev]
|
||
|
|
|
||
|
|
WORKDIR /home/petals/
|
||
|
|
CMD bash
|