From 0d8eacf6f6490e8a5f0af47a9125c33b8af63792 Mon Sep 17 00:00:00 2001
From: Jan Lukas Steimann <jan.steimann@hhu.de>
Date: Tue, 29 Jun 2021 06:05:19 +0000
Subject: [PATCH] Update .gitlab-ci.yml
---
.gitlab-ci.yml | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 64f37a7..5e4900a 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"
--
GitLab