diff --git a/recognition/arcface_onnx.py b/recognition/arcface_onnx.py
index 4b563e5..5cda3ec 100644
--- a/recognition/arcface_onnx.py
+++ b/recognition/arcface_onnx.py
@@ -6,6 +6,7 @@
 
 import numpy as np
 import cv2
+import torch # hack to use onnxruntime with cuda
 import onnx
 import onnxruntime
 import face_align
diff --git a/recognition/main.py b/recognition/main.py
index 9ba74ae..bddf777 100644
--- a/recognition/main.py
+++ b/recognition/main.py
@@ -5,6 +5,7 @@ import os.path as osp
 import argparse
 import cv2
 import numpy as np
+import torch # hack to use onnxruntime with cuda
 import onnxruntime
 from scrfd import SCRFD
 from arcface_onnx import ArcFaceONNX
diff --git a/recognition/scrfd.py b/recognition/scrfd.py
index 0b79556..2a1a156 100644
--- a/recognition/scrfd.py
+++ b/recognition/scrfd.py
@@ -2,7 +2,7 @@
 from __future__ import division
 import datetime
 import numpy as np
-#import onnx
+import torch # hack to use onnxruntime with cuda
 import onnxruntime
 import os
 import os.path as osp
diff --git a/refacer.py b/refacer.py
index 5be6d70..97795cb 100644
--- a/refacer.py
+++ b/refacer.py
@@ -1,11 +1,11 @@
 import cv2
+import torch # hack to use onnxruntime with cuda
 import onnxruntime as rt
 import sys
 from insightface.app import FaceAnalysis
 sys.path.insert(1, './recognition')
 from scrfd import SCRFD
 from arcface_onnx import ArcFaceONNX
-import os.path as osp
 import os
 from pathlib import Path
 from tqdm import tqdm