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