From d1030654cedfac9ded868ab6aee6f590c24ead2f Mon Sep 17 00:00:00 2001 From: Somdev Sangwan Date: Tue, 30 May 2023 18:39:17 +0530 Subject: [PATCH] add onnxruntime-gpu --- requirements.txt | 1 + 1 file changed, 1 insertion(+) diff --git a/requirements.txt b/requirements.txt index 14bb741..4c40f8b 100644 --- a/requirements.txt +++ b/requirements.txt @@ -6,3 +6,4 @@ psutil==5.9.5 tk==0.1.0 pillow==9.0.1 torch==2.0.1 +onnxruntime-gpu==1.15.0