Dockerfile 206 B

123456789101112
  1. ARG FROM_IMAGE_NAME=nvcr.io/nvidia/mxnet:22.10-py3
  2. FROM $FROM_IMAGE_NAME
  3. WORKDIR /workspace/rn50
  4. COPY requirements.txt .
  5. RUN pip install -r requirements.txt
  6. ENV MXNET_CUDNN_AUTOTUNE_DEFAULT=0
  7. COPY . .