Fix CUDAToolkit_ROOT

.github/workflows/win-gpu-x64-build.yml

@@ -11,6 +11,7 @@ env:
   ort_dir: "onnxruntime-win-x64-gpu-1.16.3"
   ort_zip: "onnxruntime-win-x64-gpu-1.16.3.zip"
   ort_url: "https://github.com/microsoft/onnxruntime/releases/download/v1.16.3/onnxruntime-win-x64-gpu-1.16.3.zip"
+  CUDAToolkit_ROOT: "${{ github.workspace }}\\cuda_sdk\\v12.2"
 
 jobs:
   Windows-CUDA-12-Build:
@@ -25,7 +26,13 @@ jobs:
           architecture: 'x64'
 
       - name: Download cuda
-        run: azcopy.exe cp --recursive "https://lotusscus.blob.core.windows.net/models/cuda_sdk/v12.2" cuda_sdk
+        run: |
+          azcopy.exe cp --recursive "https://lotusscus.blob.core.windows.net/models/cuda_sdk/v12.2" ${{ github.workspace }}\cuda_sdk
+
+      - name: Verify Cuda SDK
+        if: always()
+        run: |
+          echo "${{ env.CUDA_TOOLKIT_ROOT_DIR }}"
 
       - name: Download OnnxRuntime
         run: |
@@ -47,7 +54,7 @@ jobs:
         run: |
           mkdir build
           cd build
-          cmake -G "Visual Studio 17 2022" -A x64 .. -DCMAKE_BUILD_TYPE=Release -DUSE_CUDA=TRUE 
+          cmake -G "Visual Studio 17 2022" -A x64 .. -DCMAKE_BUILD_TYPE=Release -DUSE_CUDA=TRUE -DCUDAToolkit_ROOT=${{ env.CUDAToolkit_ROOT }}
           cmake --build . --config Release --parallel
 
       - name: Verify Build Artifacts

CMakeLists.txt

@@ -11,9 +11,11 @@ set(USE_ORT_EXT 0) # "Build with Onnxruntime Extensions tokenizer support"
 
 # Checking if CUDA is supported
 include(CheckLanguage)
-check_language(CUDA)
-if (CMAKE_CUDA_COMPILER)
+find_package(CUDAToolkit)
+if (CUDAToolkit_FOUND)
     set (CUDA_FOUND TRUE)
+else()
+    message(STATUS "CUDA not found")
 endif()
 
 if("${CMAKE_C_COMPILER_ID}" STREQUAL "GNU" AND CMAKE_C_COMPILER_VERSION VERSION_LESS 9)