You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I'm currently using systems run on onnxruntime 1.6.0, cuda 10.2 and cudnn 8.
How can i set "cudnn_conv_algo_search"?
I have tried the declare the same as the docs: "self.session = onnxruntime.InferenceSession(self.model_file, providers=[('CUDAExecutionProvider', {"cudnn_conv_algo_search": "HEURISTIC"}), 'CPUExecutionProvider'])" but i got the error:
Traceback (most recent call last):
File "run.py", line 5, in
app = FaceAnalysis()
File "/app/insightface/face_analysis.py", line 19, in init
self.models['detection'] = RetinaFace(mods)
File "/app/insightface/retinaface.py", line 38, in init
self.session = onnxruntime.InferenceSession(self.model_file, providers=[('CUDAExecutionProvider', {"cudnn_conv_algo_search": "HEURISTIC"}), 'CPUExecutionProvider'])
File "/usr/local/lib/python3.6/dist-packages/onnxruntime/capi/onnxruntime_inference_collection.py", line 206, in init
self._create_inference_session(providers, provider_options)
File "/usr/local/lib/python3.6/dist-packages/onnxruntime/capi/onnxruntime_inference_collection.py", line 231, in _create_inference_session
sess.initialize_session(providers or [], provider_options or [])
TypeError: initialize_session(): incompatible function arguments. The following argument types are supported:
1. (self: onnxruntime.capi.onnxruntime_pybind11_state.InferenceSession, arg0: List[str], arg1: List[Dict[str, str]]) -> None
reacted with thumbs up emoji reacted with thumbs down emoji reacted with laugh emoji reacted with hooray emoji reacted with confused emoji reacted with heart emoji reacted with rocket emoji reacted with eyes emoji
-
I'm currently using systems run on onnxruntime 1.6.0, cuda 10.2 and cudnn 8.
How can i set "cudnn_conv_algo_search"?
I have tried the declare the same as the docs: "self.session = onnxruntime.InferenceSession(self.model_file, providers=[('CUDAExecutionProvider', {"cudnn_conv_algo_search": "HEURISTIC"}), 'CPUExecutionProvider'])" but i got the error:
Traceback (most recent call last):
File "run.py", line 5, in
app = FaceAnalysis()
File "/app/insightface/face_analysis.py", line 19, in init
self.models['detection'] = RetinaFace(mods)
File "/app/insightface/retinaface.py", line 38, in init
self.session = onnxruntime.InferenceSession(self.model_file, providers=[('CUDAExecutionProvider', {"cudnn_conv_algo_search": "HEURISTIC"}), 'CPUExecutionProvider'])
File "/usr/local/lib/python3.6/dist-packages/onnxruntime/capi/onnxruntime_inference_collection.py", line 206, in init
self._create_inference_session(providers, provider_options)
File "/usr/local/lib/python3.6/dist-packages/onnxruntime/capi/onnxruntime_inference_collection.py", line 231, in _create_inference_session
sess.initialize_session(providers or [], provider_options or [])
TypeError: initialize_session(): incompatible function arguments. The following argument types are supported:
1. (self: onnxruntime.capi.onnxruntime_pybind11_state.InferenceSession, arg0: List[str], arg1: List[Dict[str, str]]) -> None
Invoked with: <onnxruntime.capi.onnxruntime_pybind11_state.InferenceSession object at 0x7fc5a1e0df80>, [('CUDAExecutionProvider', {'cudnn_conv_algo_search': 'HEURISTIC'}), 'CPUExecutionProvider'], []
Beta Was this translation helpful? Give feedback.
All reactions