diff --git a/install.js b/install.js
index fa4818c..b98172c 100644
--- a/install.js
+++ b/install.js
@@ -61,6 +61,18 @@ module.exports = async kernel =>
 					}
 				}
 			},
+			{
+				when: '{{ gpu === "nvidia" }}',
+				method: 'shell.run',
+				params:
+				{
+					message: 'pip install tensorrt==10.4.0 --extra-index-url https://pypi.nvidia.com --yes',
+					conda:
+					{
+						path: '.env'
+					}
+				}
+			},
 			{
 				method: 'shell.run',
 				params: