Update Dockerfile

Dockerfile

@@ -3,6 +3,9 @@ FROM nvidia/cuda:11.3.1-base-ubuntu20.04
 ENV DEBIAN_FRONTEND=noninteractive \
 	TZ=Europe/Paris
 
+# there is no gpu access during docker build time :(
+ENV TORCH_CUDA_ARCH_LIST Turing
+
 # Remove any third-party apt sources to avoid issues with expiring keys.
 # Install some basic utilities
 RUN rm -f /etc/apt/sources.list.d/*.list && \
@@ -83,6 +86,8 @@ USER user
 RUN --mount=target=requirements.txt,source=requirements.txt \
     pip install --no-cache-dir --upgrade -r requirements.txt
 
+RUN pip install git+https://github.com/osbm/quant_cuda
+
 # Copy the current directory contents into the container at $HOME/app setting the owner to the user
 COPY --chown=user . $HOME/app