File size: 282 Bytes
8d9de2d 8edb126 8d9de2d 8edb126 8d9de2d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
FROM pytorch/pytorch:1.9.1-cuda11.1-cudnn8-runtime
WORKDIR /usr/src/app
RUN apt-get update
RUN apt-get install libgl1 -y
Run apt-get install git -y
COPY requirements.txt ./
RUN pip install --no-cache-dir -r requirements.txt
RUN mkdir -p /.cache && chmod 777 /.cache
COPY . .
|