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