From c9e71fbc3a434eab5209c31735d72fd7e60876f2 Mon Sep 17 00:00:00 2001 From: Richard Edgar Date: Wed, 22 May 2024 13:16:38 -0400 Subject: [PATCH] [Build] Update for new CUDA Toolkit (#847) We have a new version of the CUDA Toolkit, so need to update the path we use. Without this, `cmake` wound up unable to find the toolkit when installing `llama-cpp-python`. --- .github/workflows/action_gpu_unit_tests.yml | 2 +- .github/workflows/ci_tests.yml | 2 +- .github/workflows/notebook_tests.yml | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/.github/workflows/action_gpu_unit_tests.yml b/.github/workflows/action_gpu_unit_tests.yml index 53eda0f03..0f316a115 100644 --- a/.github/workflows/action_gpu_unit_tests.yml +++ b/.github/workflows/action_gpu_unit_tests.yml @@ -42,7 +42,7 @@ jobs: - name: Ensure NVIDIA SDK available run: | sudo apt-get -y install cuda-toolkit - echo "/usr/local/cuda-12.4/bin" >> $GITHUB_PATH + echo "/usr/local/cuda-12.5/bin" >> $GITHUB_PATH - name: Install dependencies shell: bash run: | diff --git a/.github/workflows/ci_tests.yml b/.github/workflows/ci_tests.yml index f5631ab39..43c1a32b9 100644 --- a/.github/workflows/ci_tests.yml +++ b/.github/workflows/ci_tests.yml @@ -43,7 +43,7 @@ jobs: - name: Ensure NVIDIA SDK available run: | sudo apt-get -y install cuda-toolkit - echo "/usr/local/cuda-12.4/bin" >> $GITHUB_PATH + echo "/usr/local/cuda-12.5/bin" >> $GITHUB_PATH - name: Install dependencies run: | python -m pip install --upgrade pip diff --git a/.github/workflows/notebook_tests.yml b/.github/workflows/notebook_tests.yml index 174cefaa9..778816fd0 100644 --- a/.github/workflows/notebook_tests.yml +++ b/.github/workflows/notebook_tests.yml @@ -46,7 +46,7 @@ jobs: - name: Ensure NVIDIA SDK available run: | sudo apt-get -y install cuda-toolkit - echo "/usr/local/cuda-12.4/bin" >> $GITHUB_PATH + echo "/usr/local/cuda-12.5/bin" >> $GITHUB_PATH - name: Install dependencies run: | python -m pip install --upgrade pip