diff --git a/README.md b/README.md index 5db9a591e2..d4fff8bf32 100644 --- a/README.md +++ b/README.md @@ -81,7 +81,7 @@ These are the following dependencies used to verify the testcases. TRTorch can w - Libtorch 1.9.0 (built with CUDA 11.1) - CUDA 11.1 (10.2 on Jetson) - cuDNN 8.2 -- TensorRT 8.0.1.6 +- TensorRT 8.0.3.4 (TensorRT 8.0.1.6 on Jetson) ## Prebuilt Binaries and Wheel files diff --git a/WORKSPACE b/WORKSPACE index 61c791f810..259c502176 100644 --- a/WORKSPACE +++ b/WORKSPACE @@ -80,10 +80,10 @@ http_archive( http_archive( name = "tensorrt", build_file = "@//third_party/tensorrt/archive:BUILD", - sha256 = "def6a5ee50bed25a68a9c9e22ec671a8f29ee5414bde47c5767bd279e5596f88", - strip_prefix = "TensorRT-8.0.1.6", + sha256 = "3177435024ff4aa5a6dba8c1ed06ab11cc0e1bf3bb712dfa63a43422f41313f3", + strip_prefix = "TensorRT-8.0.3.4", urls = [ - "https://developer.nvidia.com/compute/machine-learning/tensorrt/secure/8.0.1/tars/tensorrt-8.0.1.6.linux.x86_64-gnu.cuda-11.3.cudnn8.2.tar.gz", + "https://developer.nvidia.com/compute/machine-learning/tensorrt/secure/8.0.3/tars/tensorrt-8.0.3.4.linux.x86_64-gnu.cuda-11.3.cudnn8.2.tar.gz", ], )