Docker file for pytorch and computer vision task

Hello. This is my first time to use the docker file and docker stuff.
My problem is that I cannot use the nvidia docker by my computer. I usually use the miniconda from school computer but school computer does not allow to use the docker and nvidia docer so I can not test it. I follow the
detectron and add some of my stuff for it.


Does anyone can figure it out it is ok?
I have to provide docker file to cloud gpu company and they will provide install the specific library which I want to use. So I hope it is ok.
I also recommend to my friend this docker file
“”
FROM nvcr.io/nvidia/pytorch:19.12-py3
USER root

RUN apt-get update && apt-get install -y \
curl
ca-certificates
sudo
unzip
htop
wget
git
bzip2
libx11-6
&& rm -rf /var/lib/apt/lists/*
RUN apt-get update -y
“”
This is official nvidia docker he said.
I can not test my computer so I am not sure they include the opencv and albumentations.
Thank you.