From e459203886ca40436046c39352d70cf32f0b524e Mon Sep 17 00:00:00 2001
From: Vlad-Andrei Badoiu <vlad_andrei.badoiu@upb.ro>
Date: Mon, 29 Jan 2024 15:10:05 +0000
Subject: [PATCH] Add gitlab CI/CD file

---
 .gitlab-ci.yml | 9 +++++++++
 1 file changed, 9 insertions(+)
 create mode 100644 .gitlab-ci.yml

diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
new file mode 100644
index 0000000..4ba1fec
--- /dev/null
+++ b/.gitlab-ci.yml
@@ -0,0 +1,9 @@
+stages:
+    - test_gpu
+
+useGPU:
+    image: pytorch/pytorch:latest
+    stage: test_gpu
+    script:
+        - echo "Check whether we have enabled our GPU or not."
+        - nvidia-smi
-- 
GitLab