| 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 . . | |