File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -140,8 +140,6 @@ jobs:
140140 run : |
141141 echo "INSTALL_ROOT=$(.venv/bin/python -c 'import cuda; print(cuda.__path__[0])')/.." >> $GITHUB_ENV
142142 echo "REPO_ROOT=$(pwd)" >> $GITHUB_ENV
143- echo "Install root: $INSTALL_ROOT"
144- echo "Repo root: $REPO_ROOT"
145143
146144 - name : Run cuda.pathfinder tests
147145 continue-on-error : true
@@ -195,16 +193,24 @@ jobs:
195193 ARCH : " amd64"
196194 CUDA_PYTHON_COVERAGE : " 1"
197195 steps :
198- - name : Ensure GPU is working
199- run : nvidia-smi
200-
201196 - name : Checkout ${{ github.event.repository.name }}
202197 uses : actions/checkout@de0fac2e4500dabe0009e67214ff5f5447ce83dd # v6.0.2
203198
204199 - name : Setup proxy cache
205200 uses : nv-gha-runners/setup-proxy-cache@main
206201 continue-on-error : true
207202
203+ - name : Update driver
204+ env :
205+ DRIVER_MODE : " TCC"
206+ GPU_TYPE : " a100"
207+ run : |
208+ ci/tools/install_gpu_driver.ps1
209+
210+ - name : Ensure GPU is working
211+ run : |
212+ nvidia-smi
213+
208214 - name : Set environment variables
209215 shell : bash --noprofile --norc -xeuo pipefail {0}
210216 env :
You can’t perform that action at this time.
0 commit comments