Windows GitHub runner nvml install

Author: 2kai2kai2

.github/workflows/npm-publish.yml

@@ -12,25 +12,32 @@ jobs:
     strategy:
       matrix:
         include:
-          - image: ubuntu-latest
-            kernel: linux-x64
+          # - image: ubuntu-latest
+          #   kernel: linux-x64
           - image: windows-latest
             kernel: win32-x64
-          - image: macos-13
-            kernel: darwin-x64
-          - image: macos-latest
-            kernel: darwin-arm64
+          # - image: macos-13
+          #   kernel: darwin-x64
+          # - image: macos-latest
+          #   kernel: darwin-arm64
     name: Build ${{ matrix.kernel }}
     runs-on: ${{ matrix.image }}
     steps:
       - uses: actions/checkout@v3
-      - uses: actions/setup-node@v2
+      - uses: actions/setup-node@v4
         with:
           node-version: '16'
       - run: |
           sudo apt-get install -y nvidia-driver-535
           sudo apt-get install -y libnvidia-ml-dev
         if: matrix.kernel == 'linux-x64'
+      - uses: conda-incubator/setup-miniconda@v3
+        with:
+          activate-environment: nvml-tools
+        if: matrix.kernel == 'win32-x64'
+      - run: |
+          conda install cuda-nvml-dev -c nvidia/label/cuda-12.9.1
+        if: matrix.kernel == 'win32-x64'
       - run: npm run install-dependencies
       - run: npm run prebuild
       - uses: actions/upload-artifact@v4

binding.gyp

@@ -10,15 +10,27 @@
                     ],
                     "include_dirs": [
                         "./node_modules/node-addon-api",
-                        "C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v12.9/include"
+                        "<!(echo %CUDA_PATH%)/include",
+                        "<!(echo %CONDA_PREFIX%)/Library/include",
                     ],
                     "ldflags": [
                         "-lpsapi", # gcc/g++
                         "-lnvidia-ml",
                     ],
                     "link_settings": {
                         "libraries": [
-                            "psapi.lib" # msvc
+                            "psapi.lib", # msvc
+                        ],
+                        "variables": {
+                            'has_cuda_path': '<!(if defined CUDA_PATH (echo 1) else (echo 0))',
+                        },
+                        "conditions": [
+                            ['has_cuda_path==1', { # locally
+                                "libraries": ["<!(echo %CUDA_PATH%)/lib/x64/nvml.lib"],
+                            }],
+                            ['has_cuda_path==0', { # github actions conda
+                                "libraries": ["<!(echo %CONDA_PREFIX%)/Library/lib/nvml.lib"],
+                            }],
                         ]
                     },
                     "cflags!": [ "-fno-exceptions" ],