Skip to content

Commit d3f48a6

Browse files
author
Chris Maunder
committed
Merge branch 'v2.9.0' of https://github.com/codeproject/CodeProject.AI-Server into v2.9.0
2 parents f572898 + c05b3a4 commit d3f48a6

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

src/setup.bat

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -417,16 +417,22 @@ REM Test for CUDA drivers
417417
call "!utilsScript!" Write "CUDA Present..."
418418

419419
set hasCUDA=false
420+
set hasCUDAToolkit=false
420421

421422
call "!utilsScript!" GetCudaVersion
422423
if "!cuda_version!" neq "" set hasCUDA=true
423424

424425
if /i "!hasCUDA!" == "true" (
426+
427+
REM CUDA Toolkit != CUDA drivers. We need the files and CUDA_PATH to be in place
428+
if exist "C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v*" set hasCUDAToolkit=true
429+
if "%CUDA_PATH%" == "" ( set "hasCUDAToolkit=false" ) else ( set "hasCUDAToolkit=true" )
430+
425431
call "!utilsScript!" GetCuDNNVersion
426432
if "!cuDNN_version!" == "" (
427-
call "!utilsScript!" WriteLine "Yes (CUDA !cuda_version!, No cuDNN found)" !color_success!
433+
call "!utilsScript!" WriteLine "Yes (CUDA !cuda_version!, No cuDNN found, CUDA Toolkit: !hasCUDAToolkit!)" !color_success!
428434
) else (
429-
call "!utilsScript!" WriteLine "Yes (CUDA !cuda_version!, cuDNN !cuDNN_version!)" !color_success!
435+
call "!utilsScript!" WriteLine "Yes (CUDA !cuda_version!, cuDNN !cuDNN_version!, CUDA Toolkit: !hasCUDAToolkit!)" !color_success!
430436
)
431437
) else (
432438
call "!utilsScript!" WriteLine "No" !color_warn!

0 commit comments

Comments
 (0)