8 lines
285 B
Plaintext
8 lines
285 B
Plaintext
FROM --platform=linux/amd64 ubuntu:22.04
|
|
|
|
RUN apt-get update -y \
|
|
&& apt-get install -y python3 python3-pip git vim wget htop numactl \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
|
|
RUN pip install scipy
|