From 72350715d3741ba3154b1d3111e41fce6cb935e6 Mon Sep 17 00:00:00 2001
From: Jan Lukas Steimann <jan.steimann@hhu.de>
Date: Tue, 29 Jun 2021 07:10:36 +0000
Subject: [PATCH] Use torch version 1.7.1
---
.gitlab-ci.yml | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index e5a97fb..608a181 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -13,8 +13,8 @@ python_run:
- sh -c "sed -i 's/main/main contrib non-free/g' /etc/apt/sources.list"
- apt update && DEBIAN_FRONTEND=noninteractive apt --no-install-recommends -y install nvidia-cuda-toolkit
- apt clean && apt autoremove && rm -rf /var/lib/apt/lists/*
+ - pip install 1.7.1+cu92 -f https://download.pytorch.org/whl/cu100/torch_stable.html
- pip install flair
- - pip install torch==1.4.0 -f https://download.pytorch.org/whl/cu100/torch_stable.html
- pip install pandas
- ./mc cp S3Uni/steimann/CommentsApril2017.csv .
- python3 ComputeCommentEmbeddings.py
--
GitLab