Breaking News: Grepper is joining You.com. Read the official announcement!

AttributeError: 'tensorrt.tensorrt.Builder' object has no attribute 'build_cuda_engine'

Merwanski answered on December 2, 2021 Popularity 3/10 Helpfulness 8/10

answer AttributeError: 'tensorrt.tensorrt.Builder' object has no attribute 'build_cuda_engine'

AttributeError: 'tensorrt.tensorrt.Builder' object has no attribute 'build_cuda_engine'

Comment

print('[info] Building an engine...')
if TRT_VERSION_MAJOR >= 7:
  # Create an optimization profile (see Section 7.2 of https://docs.nvidia.com/deeplearning/sdk/pdf/TensorRT-Developer-Guide.pdf).
  profile = builder.create_optimization_profile()
  # FIXME: Hardcoded for ImageNet. The minimum/optimum/maximum dimensions of a dynamic input tensor are the same.
  # profile.set_shape(input_tensor_name, (1, 3, 224, 224), (max_batch_size, 3, 224, 224), (max_batch_size, 3, 224, 224))

  config = builder.create_builder_config()
  config.add_optimization_profile(profile)

  trt_model_engine  = builder.build_engine(network, config)
  trt_model_context = trt_model_engine.create_execution_context()

else:
  # if trt version is below 7 then use the build_cuda_engine
  trt_model_engine  = builder.build_cuda_engine(network)
  trt_model_context = trt_model_engine.create_execution_context()

Popularity 3/10 Helpfulness 8/10 Language python

Source: Grepper

Tags: object python

Link to this answer
Share Copy Link

Contributed on Dec 02 2021

Merwanski

0 Answers Avg Quality 2/10

Continue with Google

By continuing, I agree that I have read and agree to Greppers's Terms of Service and Privacy Policy.

AttributeError: 'tensorrt.tensorrt.Builder' object has no attribute 'build_cuda_engine'

Contents

More Related Answers

AttributeError: 'tensorrt.tensorrt.Builder' object has no attribute 'build_cuda_engine'

Grepper

Documentation

Social

Legal

Contact

Oops, You will need to install Grepper and log-in to perform this action.