8 lines
189 B
Python
8 lines
189 B
Python
import onnxruntime
|
|
|
|
use_gpu = False
|
|
providers = onnxruntime.get_available_providers()
|
|
|
|
if 'TensorrtExecutionProvider' in providers:
|
|
providers.remove('TensorrtExecutionProvider')
|