Update Dockerfile

Dockerfile

@@ -1,8 +1,7 @@
 # syntax=docker/dockerfile:1
 # Build as `docker build . -t localgpt`, requires BuildKit.
 # Run as `docker run -it --mount src="$HOME/.cache",target=/root/.cache,type=bind --gpus=all localgpt`, requires Nvidia container toolkit.
-
-ARG CUDA_IMAGE="12.1.1-devel-ubuntu22.04"
+ARG CUDA_IMAGE="12.2.0-devel-ubuntu20.04"
 FROM nvidia/cuda:${CUDA_IMAGE}
 
 RUN apt-get update && apt-get upgrade -y \
@@ -14,6 +13,7 @@ RUN apt-get update && apt-get upgrade -y \
 
 RUN apt-get update && apt-get install -y software-properties-common
 RUN apt-get install -y g++-11
+RUN apt-get install -y libpq
 
 ENV TRANSFORMERS_CACHE=/root/.cache