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
RUN cargo install cargo-udeps cargo-outdated