Skip to content

Commit 1c02aa3

Browse files
committed
Revert py39 changes, make changes to infra fork, re-enable cu121 installation
1 parent dfc0e15 commit 1c02aa3

File tree

2 files changed

+2
-4
lines changed

2 files changed

+2
-4
lines changed

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

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,6 @@ jobs:
1212
test-infra-ref: windows_builds
1313
with-rocm: false
1414
with-cpu: false
15-
python-version: "3.9"
1615

1716
build:
1817
needs: generate-matrix
@@ -27,7 +26,6 @@ jobs:
2726
pre-script: packaging/pre_build_script_windows.sh
2827
smoke-test-script: packaging/smoke_test_windows.py
2928
package-name: torch_tensorrt
30-
python-version: "3.9"
3129
name: Build torch-tensorrt whl package
3230
uses: gs-olive/test-infra/.github/workflows/build_wheels_windows.yml@windows_builds
3331
with:

packaging/pre_build_script_windows.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,5 +11,5 @@ choco install bazelisk -y
1111

1212
cat ./toolchains/ci_workspaces/WORKSPACE.win.release.tmpl | envsubst > WORKSPACE
1313

14-
# curl -o cuda_12.1.0_windows_network.exe https://developer.download.nvidia.com/compute/cuda/12.1.0/network_installers/cuda_12.1.0_windows_network.exe
15-
# ./cuda_12.1.0_windows_network.exe -s nvcc_12.1 Display.Driver -n
14+
curl -o cuda_12.1.0_windows_network.exe https://developer.download.nvidia.com/compute/cuda/12.1.0/network_installers/cuda_12.1.0_windows_network.exe
15+
./cuda_12.1.0_windows_network.exe -s nvcc_12.1 Display.Driver -n

0 commit comments

Comments
 (0)