REVA-QCAV/Dockerfile
milesial f55693b6a4 Update PyTorch container, fix data download
Former-commit-id: 7679965b6814e40a9bc7be694d30cfd4bb29b9f7
2022-02-19 02:38:49 +01:00

10 lines
230 B
Docker

FROM nvcr.io/nvidia/pytorch:22.01-py3
RUN rm -rf /workspace/*
WORKDIR /workspace/unet
ADD requirements.txt .
RUN pip install --no-cache-dir --upgrade --pre pip
RUN pip install --no-cache-dir -r requirements.txt
ADD . .