ci

Lightning-AI · May 6, 2024 · a4676c7 · a4676c7
1 parent 2ff82c6
commit a4676c7
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/.azure/gpu-integrations.yml b/.azure/gpu-integrations.yml
@@ -44,7 +44,11 @@ jobs:
       clean: all
     steps:
       - bash: |
-          echo "##vso[task.setvariable variable=CUDA_VISIBLE_DEVICES]$(DEVICES)"
+          set -ex
+          devices=$(DEVICES)
+          # overwrite and use only single device
+          device=$( python -c 'print("${devices}".split(",")[0])' )
+          echo "##vso[task.setvariable variable=CUDA_VISIBLE_DEVICES]$device"
           CUDA_version=$(nvcc --version | sed -n 's/^.*release \([0-9]\+\.[0-9]\+\).*$/\1/p')
           CUDA_version_mm="${CUDA_version//'.'/''}"
           echo "##vso[task.setvariable variable=CUDA_VERSION_MM]$CUDA_version_mm"