ARG region

FROM 763104351884.dkr.ecr.${region}.amazonaws.com/pytorch-training:1.8.1-gpu-py36-cu111-ubuntu18.04

RUN pip install ninja yacs cython matplotlib tqdm opencv-python pybind11==2.5.0 'git+https://github.com/NVIDIA/dllogger'

RUN pip --no-cache-dir --no-cache install 'git+https://github.com/NVIDIA/cocoapi#egg=pycocotools&subdirectory=PythonAPI'

RUN 	cd /root && \
	rm -rf apex && \
	git clone https://github.com/NVIDIA/apex && \
	cd apex && \
	python setup.py install --cuda_ext --cpp_ext

RUN 	cd /root && \
	rm -rf DeepLearningExamples && \
	git clone --recursive https://github.com/HerringForks/DeepLearningExamples.git && \
	cd DeepLearningExamples/PyTorch/Segmentation/MaskRCNN/pytorch/ && \
	python setup.py build develop