diff --git a/Dockerfile b/Dockerfile
index 6193f55..8f1fb41 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,12 +1,10 @@
 # Start FROM Nvidia PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch
-FROM nvcr.io/nvidia/pytorch:20.03-py3
+FROM nvcr.io/nvidia/pytorch:20.08-py3
 
 # Install dependencies
-COPY requirements.txt .
-RUN pip install -r requirements.txt gsutil
-
-# ImportError: libGL.so.1: cannot open shared object file: No such file or directory solution
-RUN apt update && apt install -y libgl1-mesa-glx
+# COPY requirements.txt .
+# RUN pip install -r requirements.txt
+RUN pip install gsutil
 
 # Create working directory
 RUN mkdir -p /usr/src/app