diff --git a/Dockerfile b/Dockerfile index 803e390..60900f5 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,7 +1,6 @@ FROM ubuntu:22.04 ADD packages.txt packages.txt RUN apt-get update -y && DEBIAN_FRONTEND=noninteractive apt-get install -y $(cat packages.txt) -RUN pip3 install --no-cache-dir torch==1.13.0 torchvision==0.14.0 torchaudio==0.13.0 ADD requirements.txt requirements.txt RUN pip3 install --no-cache-dir -r requirements.txt ADD api /emblem/api