Update Dockerfile

Dockerfile

@@ -19,6 +19,9 @@ RUN python3 -m pip install --upgrade pip pytest cmake \
 RUN apt-get update && apt-get install -y software-properties-common
 RUN apt-get install -y g++-11
 
+ENV CUDA_DOCKER_ARCH=all
+ENV LLAMA_CUBLAS=1
+
 # only copy what's needed at every step to optimize layer cache
 COPY ./requirements.txt .
 
@@ -33,5 +36,6 @@ ARG device_type=cuda
 RUN python3 ingest.py --device_type $device_type
 
 COPY . .
+
 ENV device_type=cuda
 CMD python3 run_localGPT.py --device_type $device_type