From 63702a29c448cf9512a0e1c9971ecfc10857820c Mon Sep 17 00:00:00 2001
From: Henry Ruhs <info@henryruhs.com>
Date: Sun, 22 Sep 2024 16:36:40 +0200
Subject: [PATCH] Add TensorRT support

---
 install.js | 12 ++++++++++++
 1 file changed, 12 insertions(+)

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: