fix accidentally removed autoinit for pycuda

TUI-NICR · Dec 8, 2020 · d132c41 · d132c41
1 parent c449b36
commit d132c41
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/inference_time_whole_model.py b/inference_time_whole_model.py
@@ -300,6 +300,7 @@ def time_inference_onnxruntime(onnx_filepath,
 
  if args.time_tensorrt:
  import tensorrt as trt
+ import pycuda.autoinit
  import pycuda.driver as cuda
 
  print('TensorRT version:', trt.__version__)