ampere_research/pytorch/pytorch-altra.Containerfile

13 lines
449 B
Plaintext

#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 \
&& 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