FROM nvcr.io/nvidia/pytorch:22.11-py3

RUN pip install torch-pruning==1.3.5

ADD . /workspace

WORKDIR /workspace

ENV IMAGENET_ROOT /Imagenet

CMD ["sh","run_all.sh"]