ampere_research/pytorch/pytorch-epyc_7313p.Containerfile

8 lines
272 B
Plaintext
Raw Normal View History

2024-12-09 10:57:15 -05:00
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