Adding GPU check

Signed-off-by: Eric Kerfoot <[email protected]>

Author: ericspod

.github/workflows/cicd_tests.yml

@@ -66,13 +66,7 @@ jobs:
         # Test Python versions under Ubuntu with lowest PyTorch version supported
         - os: ubuntu-latest
           pytorch-version: '2.5.1'
-          python-version: '3.10'
-        - os: ubuntu-latest
-          pytorch-version: '2.5.1'
-          python-version: '3.11'
-        - os: ubuntu-latest
-          pytorch-version: '2.5.1'
-          python-version: '3.12'
+          python-version: ['3.10', '3.11', '3.12']
 
         # Test PyTorch versions under Ubuntu with lowest Python version supported
         - os: ubuntu-latest
@@ -108,16 +102,21 @@ jobs:
       with:
         path: ${{ steps.pip-cache.outputs.dir }}
         key: ${{ matrix.os }}-latest-pip-${{ steps.pip-cache.outputs.datew }}
-    - name: Install the dependencies with PyTorch ${{ matrix.pytorch-version }}
+    - name: Install the minimum dependencies
       run: |
-        /usr/bin/env nvidia-smi 2> /dev/null || echo "No nvidia-smi found"
         # min. requirements
         python -m pip install torch==${{ matrix.pytorch-version }}
         python -m pip install -r requirements-min.txt
         python -m pip list
         BUILD_MONAI=0 python setup.py develop  # no compile of extensions
       shell: bash
-    - name: Run quick tests (CPU ${{ runner.os }})
+    - if: matrix.os == 'linux-gpu-runner'
+      name: Print GPU Info
+      run: |
+        nvidia-smi
+        python -c 'import torch; print(torch.rand(2,2).to("cuda:0"))'
+      shell: bash
+    - name: Run quick tests
       run: |
         python -c 'import torch; print(torch.__version__); print(torch.rand(5,3))'
         python -c "import monai; monai.config.print_config()"
@@ -174,7 +173,7 @@ jobs:
       run: |
         python -m pip install --pre -U itk
         find  /opt/hostedtoolcache/* -maxdepth 0 ! -name 'Python' -exec rm -rf {} \;
-    - name: Install the dependencies
+    - name: Install the complete dependencies
       run: |
         python -m pip install --user --upgrade pip wheel
         python -m pip install torch==2.5.1 torchvision==0.20.1
@@ -188,7 +187,7 @@ jobs:
         python setup.py develop --uninstall
         BUILD_MONAI=1 python setup.py develop  # compile the cpp extensions
       shell: bash
-    - name: Run quick tests (CPU ${{ runner.os }})
+    - name: Run quick tests
       run: |
         python -c 'import torch; print(torch.__version__); print(torch.rand(5,3))'
         python -c "import monai; monai.config.print_config()"