FROM --platform=linux/amd64 ubuntu:22.04

RUN apt-get update -y \
	&& apt-get install -y python3 python3-pip git vim wget \
	&& rm -rf /var/lib/apt/lists/*
RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
RUN pip install scipy