Add TensorRT support

2026-07-23 06:20:48 +02:00 · 2024-09-22 16:36:40 +02:00
parent 08458cd220
commit 63702a29c4
1 changed files with 12 additions and 0 deletions
@@ -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: