diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index e5a97fbe9920ff1ede76b14b5d9c5a6c0288bd1e..608a1812590770ec8fba36f04207646657faf38a 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