update tensorrt dep docstring

README.md

@@ -74,6 +74,7 @@ sudo apt install libboost-all-dev
 ```
 git clone https://github.com/HorizonRobotics/alf
 cd alf
+pip install pybind11
 pip install -e . --extra-index-url https://download.pytorch.org/whl/cu118
 ```
 

alf/utils/tensorrt_utils.py

@@ -41,12 +41,9 @@
 
 # How to install dependencies (in a virtual env) on the deployment machine:
 # ```bash
-#   pip install onnx
+#   pip install onnx>=1.16.2 protobuf==3.20.2
 #
-#   pip install tensorrt
-# After this, tensorrt .so files will be installed in your virtual env: .../site-packages/tensorrt_lib/
-# You need to copy all .so files under it to /usr/local/cuda/targets/x86_64-linux/lib/
-# Or alternatively, you can install it system-wide from https://developer.nvidia.com/tensorrt-getting-started
+#   pip install tensorrt>=10.0
 
 # For cuda 11.x,
 #   pip install onnxruntime-gpu
@@ -61,6 +58,11 @@
 # ['TensorrtExecutionProvider', 'CUDAExecutionProvider', 'CPUExecutionProvider']
 # The order of elements represents the default priority order of Execution Providers from highest to lowest.
 
+# NOTE: If onnxruntime complains about not finding libnvinfer.so or other tensorrt libs,
+# they can be found in your virtual env: .../site-packages/tensorrt_lib/
+# You need to copy all .so files under it to /usr/local/cuda/targets/x86_64-linux/lib/
+# Or alternatively, you can install it system-wide from https://developer.nvidia.com/tensorrt-getting-started
+
 
 def is_onnxruntime_available():
     return backend is not None

setup.py

@@ -37,7 +37,7 @@
         'pathos==0.2.4',
         'pillow>=8',
         # 'procgen>=0.10.4',
-        'protobuf==3.20.1',
+        'protobuf==3.20.2',
         'psutil==5.9.8',
         'pybullet==2.5.0',
         'pyglet==1.3.2',  # higher version breaks classic control rendering