Skip to content

Commit 73d678c

Browse files
committed
fix(build): add CUDA toolkit setup for windows-gpu in workflow
1 parent b05405e commit 73d678c

1 file changed

Lines changed: 9 additions & 2 deletions

File tree

.github/workflows/main.yml

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -258,6 +258,15 @@ jobs:
258258
with:
259259
cuda_version: ${{ env.CUDA_VERSION }}
260260

261+
- name: windows setup cuda toolkit
262+
if: matrix.name == 'windows-gpu'
263+
run: |
264+
echo "C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v${{ env.CUDA_VERSION }}\bin" | Out-File -FilePath $env:GITHUB_PATH -Encoding utf8 -Append
265+
echo "C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v${{ env.CUDA_VERSION }}\libnvvp" | Out-File -FilePath $env:GITHUB_PATH -Encoding utf8 -Append
266+
echo "CUDA_PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v${{ env.CUDA_VERSION }}" | Out-File -FilePath $env:GITHUB_ENV -Append -Encoding utf8
267+
echo "CUDA_PATH_V12_4=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v${{ env.CUDA_VERSION }}" | Out-File -FilePath $env:GITHUB_ENV -Append -Encoding utf8
268+
shell: pwsh
269+
261270
- name: cache hip toolkit
262271
if: matrix.name == 'windows-gpu'
263272
id: cache-hip
@@ -391,8 +400,6 @@ jobs:
391400
make build/llama.cpp.stamp ${{ matrix.make }} LLAMA_ARGS="--target llama"
392401
make build/llama.cpp.stamp ${{ matrix.make }} LLAMA_ARGS="--target ggml"
393402
make build/llama.cpp.stamp ${{ matrix.make }} LLAMA_ARGS="--target ggml-base"
394-
env:
395-
CUDAToolkit_ROOT: "C:/Program Files/NVIDIA GPU Computing Toolkit"
396403
397404
- name: copy backend modules to dist
398405
if: false #matrix.name == 'windows-gpu'

0 commit comments

Comments
 (0)