microsoft/onnxruntime-extensions
Publicmirrored fromhttps://github.com/microsoft/onnxruntime-extensionsAvailable
.pipelines/scripts/set_env_cuda.bat
17lines · modecode
| 1 | REM Copyright (c) Microsoft Corporation. All rights reserved. |
| 2 | REM Licensed under the MIT License. |
| 3 | |
| 4 | if exist PATH=%AGENT_TEMPDIRECTORY%\v11.8\ ( |
| 5 | set PATH=%AGENT_TEMPDIRECTORY%\v11.8\bin;%AGENT_TEMPDIRECTORY%\v11.8\extras\CUPTI\lib64;%PATH% |
| 6 | ) else ( |
| 7 | set PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\bin;C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\extras\CUPTI\lib64;%PATH% |
| 8 | ) |
| 9 | |
| 10 | @REM The default version is still cuda v11.8, because set cuda v12.2 after it |
| 11 | if exist PATH=%AGENT_TEMPDIRECTORY%\v12.2\ ( |
| 12 | set PATH=%PATH%;%AGENT_TEMPDIRECTORY%\v12.2\bin;%AGENT_TEMPDIRECTORY%\v12.2\extras\CUPTI\lib64 |
| 13 | ) else ( |
| 14 | set PATH=%PATH%;C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2\bin;C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2\extras\CUPTI\lib64 |
| 15 | ) |
| 16 | |
| 17 | set GRADLE_OPTS=-Dorg.gradle.daemon=false |
| 18 | |