From 9fe1524d02bba07b8c03cc0d684e2725f351d984 Mon Sep 17 00:00:00 2001 From: bsilvereagle Date: Sun, 2 Apr 2023 14:30:57 -0700 Subject: [PATCH] Remove pinning & use CPU whls directly Co-authored-by: Pavol Rusnak --- .devops/full.Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.devops/full.Dockerfile b/.devops/full.Dockerfile index 80752f5d4..a75bc976f 100644 --- a/.devops/full.Dockerfile +++ b/.devops/full.Dockerfile @@ -7,7 +7,7 @@ RUN apt-get update && \ RUN pip install --upgrade pip setuptools wheel \ && pip install numpy requests sentencepiece tqdm \ - && pip install torch==2.0.0+cpu -f https://download.pytorch.org/whl/torch_stable.html + && pip install torch --index-url https://download.pytorch.org/whl/cpu WORKDIR /app