add onnxruntime-gpu

2023-05-30 18:39:17 +05:30
parent d3c86fb34b
commit d1030654ce
1 changed files with 1 additions and 0 deletions
--- 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