Added dockerfiles for NVidia and AMD

Dockerfile.amd-gpu

@@ -0,0 +1,15 @@
+FROM python:3.8
+
+RUN apt update -y
+
+RUN apt install libgl1 -y # or install opencv opencv-python-headless instead
+
+RUN pip3 install torch \
+    torchvision==0.11.3 \
+    -f https://download.pytorch.org/whl/rocm4.2/torch_stable.html
+    
+COPY requirements.txt /requirements.txt
+
+RUN pip3 install -r requirements.txt
+
+ENTRYPOINT ["bash"]

Dockerfile.nvidia-gpu

@@ -0,0 +1,17 @@
+FROM python:3.8
+
+RUN apt update -y
+
+RUN apt install libgl1 -y # or install opencv opencv-python-headless instead
+
+RUN pip3 install \ 
+    torch==1.11.0+cu113 \
+    torchvision==0.12.0+cu113 \
+    torchaudio==0.11.0+cu113 \
+    -f https://download.pytorch.org/whl/cu113/torch_stable.html
+
+COPY requirements.txt /requirements.txt
+
+RUN pip3 install -r requirements.txt
+
+ENTRYPOINT ["bash"]