diff --git a/docs/examples/llm/nvidia_tensorrt.ipynb b/docs/examples/llm/nvidia_tensorrt.ipynb
index b08d424a67981bf3725591083078758fc1fdd85e..64853f12f9a65ae7ef2eee6d50afa4ff29e5063b 100644
--- a/docs/examples/llm/nvidia_tensorrt.ipynb
+++ b/docs/examples/llm/nvidia_tensorrt.ipynb
@@ -57,7 +57,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "!pip3 install tensorrt_llm -U --extra-index-url https://pypi.nvidia.com"
+    "!pip install tensorrt_llm==0.7.0 --extra-index-url https://pypi.nvidia.com --extra-index-url https://download.pytorch.org/whl/cu121"
    ]
   },
   {