File size: 1,145 Bytes
ca4fc4d |
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 |
# base image
FROM nvidia/cuda:11.3.1-devel-ubuntu20.04
# avoid warnings by switching to noninteractive
ARG DEBIAN_FRONTEND=noninteractive
# make a directory for our application
RUN mkdir -p /app
WORKDIR /app
# install system-wide dependencies
RUN apt-get -qq update && \
apt-get -qq install -y --no-install-recommends curl git python3-pip python3-dev
# Install PyTorch
RUN pip3 install --upgrade torch torchvision torchaudio
# Install APEX
RUN git clone https://github.com/NVIDIA/apex.git /app/apex
WORKDIR /app/apex
RUN git checkout 265b451de8ba9bfcb67edc7360f3d8772d0a8bea
RUN pip3 install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" --global-option="--deprecated_fused_adam" --global-option="--xentropy" --global-option="--fast_multihead_attn" ./
# Install your other dependencies...
# Copy requirements.txt and install python dependencies
COPY requirements.txt .
RUN pip3 install --no-cache-dir -r requirements.txt
# Copy the rest of the application
COPY . .
# Execute the accelerate config command
# RUN accelerate config
# Command to run when starting the container
CMD ["python3", "train.py"]
|