From c9340dba5d24eb9b2dc88f084f187be4ad4b9779 Mon Sep 17 00:00:00 2001 From: gs-olive <113141689+gs-olive@users.noreply.github.com> Date: Wed, 15 Nov 2023 12:50:12 -0800 Subject: [PATCH] chore: Upgrade `release` to Torch 2.1.1 --- README.md | 2 +- WORKSPACE | 4 ++-- py/requirements.txt | 4 ++-- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/README.md b/README.md index a7100af133..b3749ca609 100644 --- a/README.md +++ b/README.md @@ -116,7 +116,7 @@ torch.jit.save(trt_ts_module, "trt_torchscript_module.ts") # save the TRT embedd These are the following dependencies used to verify the testcases. Torch-TensorRT can work with other versions, but the tests are not guaranteed to pass. - Bazel 6.2.1 -- Libtorch 2.1.0 +- Libtorch 2.1.1 - CUDA 12.1 - cuDNN 8.9.5 - TensorRT 8.6.1 diff --git a/WORKSPACE b/WORKSPACE index 2140deaf42..73deb8506c 100644 --- a/WORKSPACE +++ b/WORKSPACE @@ -54,14 +54,14 @@ http_archive( name = "libtorch", build_file = "@//third_party/libtorch:BUILD", strip_prefix = "libtorch", - urls = ["https://download.pytorch.org/libtorch/cu121/libtorch-cxx11-abi-shared-with-deps-2.1.0%2Bcu121.zip"], + urls = ["https://download.pytorch.org/libtorch/cu121/libtorch-cxx11-abi-shared-with-deps-2.1.1%2Bcu121.zip"], ) http_archive( name = "libtorch_pre_cxx11_abi", build_file = "@//third_party/libtorch:BUILD", strip_prefix = "libtorch", - urls = ["https://download.pytorch.org/libtorch/cu121/libtorch-shared-with-deps-2.1.0%2Bcu121.zip"], + urls = ["https://download.pytorch.org/libtorch/cu121/libtorch-shared-with-deps-2.1.1%2Bcu121.zip"], ) # Download these tarballs manually from the NVIDIA website diff --git a/py/requirements.txt b/py/requirements.txt index 9a6d0eb90d..45eebecff0 100644 --- a/py/requirements.txt +++ b/py/requirements.txt @@ -1,8 +1,8 @@ numpy packaging pybind11==2.6.2 -torch==2.1.0 -torchvision==0.16.0 +torch==2.1.1 +torchvision==0.16.1 --extra-index-url https://pypi.ngc.nvidia.com tensorrt==8.6.1 pyyaml