#FROM --platform=linux/arm64 ubuntu:22.04
#FROM --platform=linux/amd64 ubuntu:22.04
FROM docker.io/amperecomputingai/pytorch:latest

RUN apt-get update -y \
	&& apt-get install -y git vim wget htop \
	&& rm -rf /var/lib/apt/lists/*
RUN pip install scipy

#RUN git clone --recursive https://github.com/AmpereComputingAI/ampere_model_library.git
#RUN chmod +x /ampere_model_library/setup_deb.sh
#RUN FORCE_INSTALL=1 bash /ampere_model_library/setup_deb.sh