FROM pytorch/pytorch:1.9.0-cuda10.2-cudnn7-runtime
RUN mkdir /app
COPY requirements.txt /app
WORKDIR /app
RUN python -m pip install - upgrade pip && pip install -r requirements.txt && rm -rf requirements.txt
FROM pytorch/pytorch:1.9.0-cuda10.2-cudnn7-runtime
RUN mkdir /app
COPY requirements.txt /app
WORKDIR /app
RUN python -m pip install - upgrade pip && pip install -r requirements.txt && rm -rf requirements.txt