From 679a70c22946d696ee1b7a6cc5b068e478296841 Mon Sep 17 00:00:00 2001 From: Wiktor Walentynowicz <wiktor.walentynowicz@pwr.edu.pl> Date: Fri, 16 Dec 2022 15:03:32 +0100 Subject: [PATCH] Change dependencies installation for CPU image. --- Dockerfile | 3 +++ requirements.txt | 2 -- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/Dockerfile b/Dockerfile index 4e8a261..9b9d606 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,5 +1,8 @@ FROM python:3.10 +RUN wget https://download.pytorch.org/whl/cpu/torch-1.13.0%2Bcpu-cp310-cp310-linux_x86_64.whl +RUN python3 -m pip install ./torch-1.13.0+cpu-cp310-cp310-linux_x86_64.whl && rm torch-1.13.0+cpu-cp310-cp310-linux_x86_64.whl + COPY requirements.txt requirements.txt RUN python3 -m pip install -r requirements.txt && rm requirements.txt diff --git a/requirements.txt b/requirements.txt index fca5f34..d3bca52 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,3 @@ ---index-url https://download.pytorch.org/whl/cpu -torch==1.13.0 --index-url https://pypi.clarin-pl.eu/simple/ nlp_ws winer==0.3.1 -- GitLab