Skip to content

Commit 78ed7d1

Browse files
committed
Test cuda-python on MCDM mode
Switch the GPU to MCDM mode and run the test suite.
1 parent d8b4acc commit 78ed7d1

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

.github/workflows/test-wheel-windows.yml

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,9 +87,15 @@ jobs:
8787
run: |
8888
.github/workflows/install_gpu_driver.ps1
8989
90-
- name: Ensure GPU is working
90+
- name: Ensure GPU is working in factory mode
9191
run: nvidia-smi
9292

93+
- name: Switch to MCDM mode
94+
run: |
95+
nvidia-smi -fdm 2
96+
pnputil /disable-device /class Display
97+
pnputil /enable-device /class Display
98+
9399
- name: Set environment variables
94100
env:
95101
BUILD_CUDA_VER: ${{ inputs.build-ctk-ver }}

0 commit comments

Comments
 (0)