diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml new file mode 100644 index 0000000000000000000000000000000000000000..4ba1fec2a6ba912a73595564662b55a7d145b589 --- /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