Update Dockerfile

Dockerfile

@@ -1,4 +1,4 @@
-FROM nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04
+FROM tensorflow/tensorflow:2.0.0-gpu-py3
 
 WORKDIR /code
 
@@ -6,8 +6,18 @@ COPY ./requirements.txt /code/requirements.txt
 
 RUN apt-get update && apt-get install -y libgl1-mesa-glx libosmesa6
 RUN apt-get install python3-opengl
-RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+RUN apt-cache search mesa
+RUN apt-get update && apt-get install -y libgl1-mesa-glx libosmesa6
 
+# Run the command inside your image filesystem
+RUN pip install --upgrade pip && \
+        pip install gym && \
+        pip install gym[atari] && \
+        pip install PyOpenGL==3.1.0 && \
+        pip install dm_control && \
+        pip install tqdm && \
+        pip install np_utils && \
+        pip install opencv-python
 COPY . .
 
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]