Skip to content

Commit c2e48a5

Browse files
committed
Upgrade CUDA version in Windows CI
1 parent 15fe3d0 commit c2e48a5

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

.github/workflows/windows.yml

+3-2
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ concurrency:
2121

2222
env:
2323
PIP_VER: "24.3.1"
24-
CUDA_VERSION: "12.1.0"
24+
CUDA_VERSION: "12.4.0"
2525
SRC_DIR: "D:\\a\\open3d\\open3d"
2626
BUILD_DIR: "C:\\Open3D\\build"
2727
NPROC: 2
@@ -106,7 +106,8 @@ jobs:
106106
- name: Install PyTorch requirements
107107
run: |
108108
if (${{ matrix.BUILD_CUDA_MODULE == 'ON' }}) {
109-
python -m pip install -r open3d_ml/requirements-torch-cuda.txt
109+
python -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
110+
python -m pip install tensorboard
110111
} else {
111112
python -m pip install -r open3d_ml/requirements-torch.txt
112113
}

0 commit comments

Comments
 (0)