Skip to content

Commit 0603d53

Browse files
committed
chore: Update pytorch version
Signed-off-by: Dheeraj Peri <[email protected]>
1 parent 9920cfd commit 0603d53

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

.circleci/config.yml

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -260,10 +260,10 @@ commands:
260260
parameters:
261261
torch-build:
262262
type: string
263-
default: "1.13.0"
263+
default: "1.14.0.dev20221114+cu117"
264264
torch-build-index:
265265
type: string
266-
default: "https://download.pytorch.org/whl/cu117"
266+
default: "https://download.pytorch.org/whl/nightly/cu117"
267267
steps:
268268
- run:
269269
name: Install Torch
@@ -428,7 +428,7 @@ commands:
428428
- run:
429429
name: Test torch
430430
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())"
432432
433433
- run:
434434
name: Get torch-tensorrt version information
@@ -792,6 +792,7 @@ jobs:
792792
TORCHTRT_VERSION=$(cd ~/project/py/torch_tensorrt && python3 -c "from _version import __version__;print(__version__)")
793793
TRT_VERSION=$(cd ~/project/py/torch_tensorrt && python3 -c "from _version import __tensorrt_version__;print(__tensorrt_version__)")
794794
CUDNN_VERSION=$(cd ~/project/py/torch_tensorrt && python3 -c "from _version import __cudnn_version__;print(__cudnn_version__)")
795+
pip3 install --upgrade pip
795796
pip3 install -r ~/project/py/requirements.txt
796797
TORCH_VERSION=$(python3 -c "from torch import __version__;print(__version__.split('+')[0])")
797798
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:
11451146
torch-build: << pipeline.parameters.torch-build >>
11461147
torch-build-index: << pipeline.parameters.torch-build-index >>
11471148
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

Comments
 (0)