sadmen
/
openpilot_comma
mirror of https://github.com/commaai/openpilot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix model runtime on PC (#34210)

Browse Source 
exhaustive
pull/34217/head
Maxime Desroches 5 months ago committed by GitHub
parent 1dcdf57395
commit 216ebcaa50
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 2
      selfdrive/modeld/runners/onnxmodel.py

2
selfdrive/modeld/runners/onnxmodel.py
Unescape View File

@ -49,7 +49,7 @@ def create_ort_session(path, fp16_to_fp32):
provider = 'OpenVINOExecutionProvider' provider = 'OpenVINOExecutionProvider'
elif 'CUDAExecutionProvider' in ort.get_available_providers() and 'ONNXCPU' not in os.environ: elif 'CUDAExecutionProvider' in ort.get_available_providers() and 'ONNXCPU' not in os.environ:
options.intra_op_num_threads = 2 options.intra_op_num_threads = 2
provider = ('CUDAExecutionProvider', {'cudnn_conv_algo_search': 'DEFAULT'}) provider = ('CUDAExecutionProvider', {'cudnn_conv_algo_search': 'EXHAUSTIVE'})
else: else:
options.intra_op_num_threads = 2 options.intra_op_num_threads = 2
options.execution_mode = ort.ExecutionMode.ORT_SEQUENTIAL options.execution_mode = ort.ExecutionMode.ORT_SEQUENTIAL

Write Preview
Loading…
Cancel
Save