DeepLearningExamples/PyTorch/Classification/ConvNets/Dockerfile

9 lines
228 B
Docker
Raw Normal View History

ARG FROM_IMAGE_NAME=nvcr.io/nvidia/pytorch:20.06-py3
2019-12-15 05:13:59 +01:00
FROM ${FROM_IMAGE_NAME}
2019-12-20 14:46:11 +01:00
ADD requirements.txt /workspace/
WORKDIR /workspace/
RUN pip install --no-cache-dir -r requirements.txt
2019-12-15 05:13:59 +01:00
ADD . /workspace/rn50
WORKDIR /workspace/rn50