From a575a27f77e63a5f912cd2c1d7ce76a1efcac67c Mon Sep 17 00:00:00 2001 From: Wiktor Walentynowicz <wiktor.walentynowicz@pwr.edu.pl> Date: Fri, 16 Dec 2022 14:46:09 +0100 Subject: [PATCH] Add different dependencies to dockerfiles. --- requirements-gpu.txt | 3 ++- requirements.txt | 4 +++- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/requirements-gpu.txt b/requirements-gpu.txt index 3508634..eefb3a3 100644 --- a/requirements-gpu.txt +++ b/requirements-gpu.txt @@ -1,4 +1,5 @@ +torch==1.13.0 --index-url https://pypi.clarin-pl.eu/simple/ nlp_ws -winer==0.3.0 +winer[torch-gpu]==0.3.1 awscli==1.22.57 \ No newline at end of file diff --git a/requirements.txt b/requirements.txt index 3508634..fca5f34 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,6 @@ +--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.0 +winer==0.3.1 awscli==1.22.57 \ No newline at end of file -- GitLab