diff --git a/README.md b/README.md index 2d2eb23805..9cf094e1aa 100644 --- a/README.md +++ b/README.md @@ -112,7 +112,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 5.1.1 -- Libtorch 1.11.0 (built with CUDA 11.3) +- Libtorch 1.12.0 (built with CUDA 11.3) - CUDA 11.3 - cuDNN 8.4.1 - TensorRT 8.4.1.5 diff --git a/WORKSPACE b/WORKSPACE index 6518b1f23c..0e5169a5e0 100644 --- a/WORKSPACE +++ b/WORKSPACE @@ -56,17 +56,17 @@ new_local_repository( http_archive( name = "libtorch", build_file = "@//third_party/libtorch:BUILD", - sha256 = "8d9e829ce9478db4f35bdb7943308cf02e8a2f58cf9bb10f742462c1d57bf287", + sha256 = "80f089939de20e68e3fcad4dfa72a26c8bf91b5e77b11042f671f39ebac35865", strip_prefix = "libtorch", - urls = ["https://download.pytorch.org/libtorch/cu113/libtorch-cxx11-abi-shared-with-deps-1.11.0%2Bcu113.zip"], + urls = ["https://download.pytorch.org/libtorch/cu113/libtorch-cxx11-abi-shared-with-deps-1.12.0%2Bcu113.zip"], ) http_archive( name = "libtorch_pre_cxx11_abi", build_file = "@//third_party/libtorch:BUILD", - sha256 = "90159ecce3ff451f3ef3f657493b6c7c96759c3b74bbd70c1695f2ea2f81e1ad", + sha256 = "8e35371403f7052d9e9b43bcff383980dbde4df028986dc1dab539953481d55f", strip_prefix = "libtorch", - urls = ["https://download.pytorch.org/libtorch/cu113/libtorch-shared-with-deps-1.11.0%2Bcu113.zip"], + urls = ["https://download.pytorch.org/libtorch/cu113/libtorch-shared-with-deps-1.12.0%2Bcu113.zip"], ) # Download these tarballs manually from the NVIDIA website diff --git a/py/requirements.txt b/py/requirements.txt index fce4b91dca..2f49390911 100644 --- a/py/requirements.txt +++ b/py/requirements.txt @@ -1,3 +1,3 @@ --extra-index-url https://download.pytorch.org/whl/cu113 -torch==1.11.0 +torch==1.12.0+cu113 pybind11==2.6.2 diff --git a/py/torch_tensorrt/ts/_compile_spec.py b/py/torch_tensorrt/ts/_compile_spec.py index 204f4cf91c..4c7b8b5b5d 100644 --- a/py/torch_tensorrt/ts/_compile_spec.py +++ b/py/torch_tensorrt/ts/_compile_spec.py @@ -350,7 +350,7 @@ def TensorRTCompileSpec(inputs=[], backend_spec._set_workspace_size(parsed_spec.workspace_size) backend_spec._set_dla_sram_size(parsed_spec.dla_sram_size) backend_spec._set_dla_local_dram_size(parsed_spec.dla_local_dram_size) - backend_spec._set_dla_global_dram_size(parsed_spec._set_dla_global_dram_size) + backend_spec._set_dla_global_dram_size(parsed_spec.dla_global_dram_size) backend_spec._set_truncate_long_and_double(parsed_spec.truncate_long_and_double) backend_spec._set_ptq_calibrator(parsed_spec._get_calibrator_handle()) diff --git a/tests/py/requirements.txt b/tests/py/requirements.txt index 784cca7d8d..0ea1c76a29 100644 --- a/tests/py/requirements.txt +++ b/tests/py/requirements.txt @@ -1,3 +1,2 @@ -torchvision==0.12.0 ---extra-index-url https://download.pytorch.org/whl/cu113 -pytest +torchvision==0.13.0+cu113 +-f https://download.pytorch.org/whl/torch_stable.html