won't use cp unless using nvidia

This commit is contained in:
Elijah Harmon 2023-11-15 08:18:34 -08:00
parent f85496859a
commit 9583a6b58a

View File

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