update Dockerfile to allow gradio

Dockerfile

@@ -1,4 +1,4 @@
-FROM nvidia/cuda:11.3.1-devel-ubuntu18.04
+FROM nvidia/cuda:11.3.1-devel-ubuntu20.04
 
 ENV PROJECT=vidar
 ENV LC_ALL=C.UTF-8
@@ -9,16 +9,18 @@ ENV PYTORCH_VERSION=1.10.0+cu113
 ENV TORCHVISION_VERSION=0.11.1+cu113
 ENV CUDNN_VERSION=8.2.1.32-1+cuda11.3
 ENV NCCL_VERSION=2.9.9-1+cuda11.3
+ARG DEBIAN_FRONTEND=noninteractive
 
 # Install basic libraries
 RUN apt-get update && apt-get install -y \
-    build-essential cmake g++-4.8 git curl docker.io vim wget ca-certificates
+    build-essential cmake g++ git curl docker.io vim wget ca-certificates
+RUN apt-get install -y --no-install-recommends apt-utils
+
 
 # Install python and pip
 RUN apt-get install -y python${PYTHON_VERSION} python3-pip 
-RUN ln -s /usr/bin/python${PYTHON_VERSION} /usr/bin/python & \
-    ln -s /usr/bin/pip3 /usr/bin/pip
-
+RUN ln -s /usr/bin/python${PYTHON_VERSION} /usr/bin/python
+    
 # Upgrade pip
 RUN pip install --upgrade pip 
 
@@ -80,7 +82,7 @@ RUN apt-get install -y \
 RUN pip install pytorch3d
 
 # Install CuPY
-RUN pip install cupy
+RUN pip install cupy-cuda11x
 
 # Install huggingface transformers
 RUN pip install transformers
@@ -95,6 +97,9 @@ WORKDIR /workspace
 RUN git clone https://github.com/VitorGuizilini-TRI/dgp.git
 ENV PYTHONPATH="/workspace/dgp:$PYTHONPATH"
 
+# Install gradio
+RUN pip install gradio
+
 # Create workspace folder
 RUN mkdir -p /workspace/experiments
 RUN mkdir -p /workspace/${PROJECT}
@@ -105,3 +110,7 @@ COPY . /workspace/${PROJECT}
 # Set environment variables
 ENV PYTHONPATH="/workspace/${PROJECT}:$PYTHONPATH"
 ENV PYTHONPATH="/workspace/${PROJECT}/externals/camviz:$PYTHONPATH"
+
+# command
+CMD ["python3", "app.py"]
+# CMD ["gradio", "app.py"] # debug

Makefile

@@ -46,7 +46,7 @@ clean:
 
 docker-build:
 	docker build \
-		-f docker/Dockerfile \
+		-f Dockerfile \
 		-t ${DOCKER_IMAGE} .
 
 docker-interactive: docker-build