barian0517/APEX_AIMBOT
1
0
Fork 0
mirror of https://github.com/RootKit-Org/AI-Aimbot.git synced 2025-06-21 02:41:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

won't use cp unless using nvidia

Browse Source
This commit is contained in:
Elijah Harmon 2023-11-15 08:18:34 -08:00
parent f85496859a
commit 9583a6b58a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
main_onnx.py
View File

@ -1,6 +1,5 @@
import onnxruntime as ort import onnxruntime as ort
import numpy as np import numpy as np
import cupy as cp
import gc import gc
import numpy as np import numpy as np
import cv2 import cv2
@ -32,6 +31,7 @@ def main():
elif onnxChoice == 2: elif onnxChoice == 2:
onnxProvider = "DmlExecutionProvider" onnxProvider = "DmlExecutionProvider"
elif onnxChoice == 3: elif onnxChoice == 3:
import cupy as cp
onnxProvider = "CUDAExecutionProvider" onnxProvider = "CUDAExecutionProvider"
so = ort.SessionOptions() so = ort.SessionOptions()