DeepLearningExamples/PyTorch/Classification/ConvNets/Dockerfile
Krzysztof Kudrynski f0ef8493eb ConvNets update
2019-12-20 14:54:58 +01:00

8 lines
228 B
Docker

ARG FROM_IMAGE_NAME=nvcr.io/nvidia/pytorch:19.10-py3
FROM ${FROM_IMAGE_NAME}
ADD requirements.txt /workspace/
WORKDIR /workspace/
RUN pip install --no-cache-dir -r requirements.txt
ADD . /workspace/rn50
WORKDIR /workspace/rn50