doc: Update installation instructions based on ABI and platforms

Signed-off-by: Dheeraj Peri <[email protected]>

Author: peri044

README.md

@@ -56,7 +56,7 @@ import torch_tensorrt
 
 ...
 
-trt_ts_module = torch_tensorrt.compile(torch_script_module, 
+trt_ts_module = torch_tensorrt.compile(torch_script_module,
     inputs = [example_tensor, # Provide example tensor for input shape or...
         torch_tensorrt.Input( # Specify input object with shape and dtype
             min_shape=[1, 3, 224, 224],
@@ -204,9 +204,10 @@ bazel build //:libtorchtrt --compilation_mode=dbg
 ```
 
 ### Native compilation on NVIDIA Jetson AGX
+We performed end to end testing on Jetson platform using Jetpack SDK 4.6. 
 
 ``` shell
-bazel build //:libtorchtrt --distdir third_party/dist_dir/aarch64-linux-gnu
+bazel build //:libtorchtrt --platforms //toolchains:jetpack_4.6
 ```
 
 > Note: Please refer [installation](docs/tutorials/installation.html) instructions for Pre-requisites

py/README.md

@@ -31,9 +31,16 @@ results = compiled_trt_model(data.half())
 
 ## Installation
 
-```
-python3 setup.py install
-```
+| ABI / Platform                          | Installation command                                         |
+| --------------------------------------- | ------------------------------------------------------------ |
+| Pre CXX11 ABI (Linux x86_64)            | python3 setup.py install --release                           |
+| CXX ABI  (Linux x86_64)                 | python3 setup.py install --use-cxx11-abi --release           |
+| Pre CXX11 ABI (Jetson platform aarch64) | python3 setup.py install --release --jetpack-version 4.6     |
+| CXX11 ABI (Jetson platform aarch64)     | python3 setup.py install --release --jetpack-version 4.6 --use-cxx11-abi |
+
+For Linux x86_64 platform, Pytorch libraries default to pre cxx11 abi. So, please use `python3 setup.py install --release`.
+
+On Jetson platforms, NVIDIA hosts <a href="https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-10-now-available/72048">pre-built Pytorch wheel files</a>. These wheel files are built with CXX11 ABI. So on jetson platforms, please use `python3 setup.py install --release --jetpack-version 4.6`   
 
 ## Under the Hood