FROM rust:1.54.0

WORKDIR /app

RUN apt update &&\

rm -rf ~/.cache &&\

apt clean all &&\

apt install -y cmake &&\

apt install -y clang

WORKDIR /

RUN wget https://download.pytorch.org/libtorch/cpu/libtorch-cxx11-abi-shared-with-deps-1.9.0%2Bcpu.zip -O libtorch.zip

RUN unzip -o libtorch.zip

ENV LIBTORCH /libtorch

ENV LD_LIBRARY_PATH /libtorch/lib:$LD_LIBRARY_PATH

RUN wget https://github.com/LaurentMazare/ocaml-torch/releases/download/v0.1-unstable/resnet18.ot -O resnet18.ot

WORKDIR /app