When I run the sample named “network_api_pytorch_mnist” by using
python sample.py -d /home/myusername/TensorRT-18.104.22.168/python/data, it will return “Segmentation fault”.I check the problem and find that the error at the code ‘import
When I run other python sample like ‘end_to_end_tensorflow_mnist’, it will return the error as followed:
[TensorRT] ERROR: runtime.cpp(24) - Cuda Error in allocate: 2 [TensorRT] ERROR: runtime.cpp(24) - Cuda Error in allocate: 2 Traceback(most recent call last): File "sample.py", line 110, in <module> main() File "sample.py", line 96, in main with build_engine(model_file) as engine: AttributeError: __exit__
I check the probloem and find the error at “builder.build_cuda_engine()” method.
How can I solve them?
Help me please.TAT