diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 8bc75dd727ce61fde7bf56c824fa400735e11a72..b4fd2a20777fedbd2e0903cb62750844d8566aed 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
-    - pip install torch
+    - 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 pandas
     - ./mc cp S3Uni/steimann/CommentsApril2017.csv .
     - python3 ComputeCommentEmbeddings.py