Fix CI

Signed-off-by: Nikolaos Karaolidis <nick@karaolidis.com>

support/ci/Dockerfile

@@ -1,4 +1,12 @@
-FROM rust
+FROM rust:bookworm
+
+RUN apt-get update -y && apt-get upgrade -y
+RUN apt-get install -y python3 python3-setuptools python3-pip
+RUN apt-get clean
+RUN rm -rf /var/lib/apt/lists/*
+
+RUN pip3 install torch==2.1.0+cpu --index-url https://download.pytorch.org/whl/cpu --break-system-packages
+ENV LIBTORCH_USE_PYTORCH 1
 
 RUN rustup install nightly
 RUN rustup component add rustfmt clippy --toolchain nightly