From 972f9d58cc021091afda709d5cf785c704ad51d6 Mon Sep 17 00:00:00 2001 From: Jan Lukas Steimann <jan.steimann@hhu.de> Date: Tue, 29 Jun 2021 07:05:55 +0000 Subject: [PATCH] Try other version of torch --- .gitlab-ci.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml index b4fd2a2..e5a97fb 100644 --- a/.gitlab-ci.yml +++ b/.gitlab-ci.yml @@ -14,7 +14,7 @@ python_run: - 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 flair - - pip3 install --upgrade torch==1.7.0+cu92 torchvision==0.8.1+cu92 torchaudio==0.7.0 -f https://download.pytorch.org/whl/torch_stable.html + - 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