@@ -260,10 +260,10 @@ commands:
260
260
parameters :
261
261
torch-build :
262
262
type : string
263
- default : " 1.13.0 "
263
+ default : " 1.14.0.dev20221114+cu117 "
264
264
torch-build-index :
265
265
type : string
266
- default : " https://download.pytorch.org/whl/cu117"
266
+ default : " https://download.pytorch.org/whl/nightly/ cu117"
267
267
steps :
268
268
- run :
269
269
name : Install Torch
@@ -428,7 +428,7 @@ commands:
428
428
- run :
429
429
name : Test torch
430
430
command : |
431
- python3 -c "import torch; print(torch.cuda.is_available()); print(torch.cuda.device_count()); print(torch.__version__) "
431
+ python3 -c "import torch; print(torch.cuda.is_available()); print(torch.cuda.device_count())"
432
432
433
433
- run :
434
434
name : Get torch-tensorrt version information
@@ -792,6 +792,7 @@ jobs:
792
792
TORCHTRT_VERSION=$(cd ~/project/py/torch_tensorrt && python3 -c "from _version import __version__;print(__version__)")
793
793
TRT_VERSION=$(cd ~/project/py/torch_tensorrt && python3 -c "from _version import __tensorrt_version__;print(__tensorrt_version__)")
794
794
CUDNN_VERSION=$(cd ~/project/py/torch_tensorrt && python3 -c "from _version import __cudnn_version__;print(__cudnn_version__)")
795
+ pip3 install --upgrade pip
795
796
pip3 install -r ~/project/py/requirements.txt
796
797
TORCH_VERSION=$(python3 -c "from torch import __version__;print(__version__.split('+')[0])")
797
798
cp ~/project/bazel-bin/libtorchtrt.tar.gz ~/project/py/dist/libtorchtrt-${TORCHTRT_VERSION}-cudnn${CUDNN_VERSION}-tensorrt${TRT_VERSION}-cuda${CUDA_VERSION}-libtorch${TORCH_VERSION}-x86_64-linux.tar.gz
@@ -1145,4 +1146,4 @@ workflows:
1145
1146
torch-build : << pipeline.parameters.torch-build >>
1146
1147
torch-build-index : << pipeline.parameters.torch-build-index >>
1147
1148
trt-version-short : << pipeline.parameters.trt-version-short >>
1148
- cudnn-version : << pipeline.parameters.cudnn-version >>
1149
+ cudnn-version : << pipeline.parameters.cudnn-version >>
0 commit comments