File tree Expand file tree Collapse file tree 1 file changed +2
-14
lines changed Expand file tree Collapse file tree 1 file changed +2
-14
lines changed Original file line number Diff line number Diff line change 1
1
python -m pip install -U numpy packaging pyyaml setuptools wheel
2
- python -c " import torch; print('IMPORTED TORCH 1')"
3
- python -m pip config set global.index-url " https://pypi.org/simple https://download.pytorch.org/whl/nightly/cu118 https://download.pytorch.org/whl/nightly/cu121"
4
- python -m pip config set global.extra-index-url " https://pypi.nvidia.com"
5
- python -m pip config set global.no-cache-dir true
6
- python -c " import torch; print('IMPORTED TORCH 2')"
7
- python -m pip install tensorrt==9.3.0.post12.dev1 tensorrt_libs==9.3.0.post12.dev1 tensorrt_bindings==9.3.0.post12.dev1
8
- python -c " import torch; print('IMPORTED TORCH 3')"
2
+ python -m pip install tensorrt==10.0.0b6 tensorrt-cu12-bindings==10.0.0b6 tensorrt-cu12-libs==10.0.0b6 --extra-index-url https://pypi.nvidia.com
9
3
10
4
choco install bazelisk -y
11
5
12
- cat ./toolchains/ci_workspaces/WORKSPACE.win.release.tmpl | envsubst > WORKSPACE
13
-
14
- # Adapted from:
15
- # https://github.com/HolyWu/TensorRT/commit/41237cedd56809062ad8e29ed0a18fabd54dd3e1#diff-97ebcd21a0f2f4e262f4e6c703e13e8cc6b233f9d9861eb2d9d06994f555f144
16
- # curl -so 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
17
- # ./cuda_12.1.0_windows_network.exe -s nvcc_12.1 Display.Driver -n
18
- # ./cuda_12.1.0_windows_network.exe -s cuda_profiler_api_12.1 cudart_12.1 cuobjdump_12.1 cupti_12.1 cuxxfilt_12.1 nvcc_12.1 nvdisasm_12.1 nvjitlink_12.1 nvml_dev_12.1 nvprof_12.1 nvprune_12.1 nvrtc_12.1 nvrtc_dev_12.1 nvtx_12.1 nvvm_samples_12.1 opencl_12.1 visual_profiler_12.1 sanitizer_12.1 thrust_12.1 cublas_12.1 cublas_dev_12.1 cufft_12.1 cufft_dev_12.1 curand_12.1 curand_dev_12.1 cusolver_12.1 cusolver_dev_12.1 cusparse_12.1 cusparse_dev_12.1 npp_12.1 npp_dev_12.1 nvjpeg_12.1 nvjpeg_dev_12.1 -n
6
+ cat toolchains/ci_workspaces/WORKSPACE.win.release.tmpl | envsubst > WORKSPACE
You can’t perform that action at this time.
0 commit comments