From 7cbffe4a6f46a6f0eae68f293f5f1c44fc8ba8af Mon Sep 17 00:00:00 2001 From: Somdev Sangwan Date: Wed, 31 May 2023 14:37:59 +0530 Subject: [PATCH] Update requirements.txt --- requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index 81b529d..4eeae19 100644 --- a/requirements.txt +++ b/requirements.txt @@ -6,4 +6,4 @@ psutil==5.9.5 tk==0.1.0 pillow==9.5.0 torch==2.0.1 -onnxruntime==1.15.0 +onnxruntime-gpu==1.15.0