Skip to content

Commit 5413339

Browse files
authored
[Build] Fix GPU builds (#1453)
The latest PyTorch disagrees with the NVIDIA driver version on the GitHub runners. For now, put an upper bound on PyTorch, but this isn't a long term solution.
1 parent 172f1d4 commit 5413339

1 file changed

Lines changed: 5 additions & 2 deletions

File tree

.github/workflows/call_gpu_tests.yml

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -63,8 +63,11 @@ jobs:
6363
run: |
6464
nvidia-smi
6565
sudo apt-get --yes update
66-
sudo apt-get --yes install cuda-toolkit-12.6
67-
echo "/usr/local/cuda-12.6/bin" >> $GITHUB_PATH
66+
sudo apt-get --yes install cuda-toolkit-12.8
67+
echo "/usr/local/cuda-12.8/bin" >> $GITHUB_PATH
68+
- name: Install older pytorch (CUDA compatibility)
69+
run: |
70+
pip install "torch<2.11"
6871
- name: Install other packages
6972
shell: bash
7073
run: |

0 commit comments

Comments
 (0)