The compiler image deployed using kubernetes does not support the compilation of checkpoint to TensorRT

[KellyZhang2020]

When compile the checkpoint model to TensorRT, the error message is as follows：

[TensorRT] ERROR: ../rtSafe/safeContext.cpp (110) - cuBLAS Error in initializeCommonContext: 1 (Could not initialize cublas, please check cuda installation.)
[TensorRT] ERROR: ../rtSafe/safeContext.cpp (110) - cuBLAS Error in initializeCommonContext: 1 (Could not initialize cublas, please check cuda installation.)
Traceback (most recent call last):
  File "/usr/local/lib/python3.6/dist-packages/model_compiler/__init__.py", line 13, in compile_model
    'path': compiler.compile_from_json(config)
  File "/usr/local/lib/python3.6/dist-packages/model_compiler/compiler.py", line 50, in compile_from_json
    target_model = compiler(source=source_type.from_json(value), config=compiler_config_type.from_json(value))
  File "/usr/local/lib/python3.6/dist-packages/model_compiler/compilers/repository.py", line 118, in _compiler
    result = edge.compiler(result, inner_config)
  File "/usr/local/lib/python3.6/dist-packages/model_compiler/compilers/onnx_model_to_tensorrt_model.py", line 90, in compile_source
    raise ValueError('Unable to build CUDA engine')
ValueError: Unable to build CUDA engine
{'status': 'failure', 'error_msg': 'Unable to build CUDA engine'}