diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 64f37a7b3d5abf60a71b2f7d074d1c52951e0bf3..5e4900a4dd265d2703ac4e3f7840132279114b60 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -7,6 +7,7 @@ before_script:
 
 python_run:
   script:
+    - apt update && apt install -y nvidia-cuda-toolkit
     - pip install flair
     - pip install torch
     - pip install pandas
@@ -14,4 +15,4 @@ python_run:
     - python3 ComputeCommentEmbeddings.py
     - ./mc cp embedded_comments.pt S3Uni/steimann/
   tags:
-    - "ht"
\ No newline at end of file
+    - "gpu"