From 3823389e3f91cab47b72e2cc2814826db9f98e32 Mon Sep 17 00:00:00 2001
From: araza008 <159492532+araza008@users.noreply.github.com>
Date: Tue, 13 Feb 2024 21:26:20 -0600
Subject: [PATCH] Update TensorRT-LLM version to 0.7.0 to resolve installation
failures (#10679)
---
docs/examples/llm/nvidia_tensorrt.ipynb | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/docs/examples/llm/nvidia_tensorrt.ipynb b/docs/examples/llm/nvidia_tensorrt.ipynb
index b08d424a67..64853f12f9 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"
]
},
{
--
GitLab