benz725 commented on issue #19731: URL: https://github.com/apache/incubator-mxnet/issues/19731#issuecomment-757467466
> > You will need to install tensorrt first. Please see https://developer.nvidia.com/Tensorrt > > Yes , I already have installed Tensorrt. It is in my /usr/local/TensorRT-7.2.1.6/ directory. @leezu and follow the https://github.com/NVIDIA/TensorRT/tree/master/samples/opensource/sampleMNIST tutorial, I run the sample_mnist program to confirm my tensorrt is installed correctly. the output of the sample_mnist is below: &&&& RUNNING TensorRT.sample_mnist # ./sample_mnist [01/10/2021-20:19:35] [I] Building and running a GPU inference engine for MNIST [01/10/2021-20:19:42] [I] [TRT] Some tactics do not have sufficient workspace memory to run. Increasing workspace size may increase performance, please check verbose output. [01/10/2021-20:19:42] [I] [TRT] Detected 1 inputs and 1 output network tensors. [01/10/2021-20:19:42] [I] Input: @@@@@@@@@@@@@@@@@@@@@@@@@@@@ @@@@@@@@@@@@@@@@@@@@@@@@@@@@ @@@@@@@@@@@@@@@@@@@@@@@@@@@@ @@@@@@@@@@@@@@@@@@@@@@@@@@@@ @@@@@@@@@@@@@@@@@@@@@@@@@@@@ @@@@@@@@@@@@@@@@@@+ #@@@@@@ @@@@@@@@@@@@@@@@@* #@@@@@@ @@@@@@@@@@@@@@@@@= #@@@@@@ @@@@@@@@@@@@@@@@# .%@@@@@@@ @@@@@@@@@@@@@@@#. *@@@@@@@@ @@@@@@@@@@@@@@@- :@@@@@@@@@ @@@@@@@@@@@@@@@ #@@@@@@@@@ @@@@@@@@@@@@@%. =@@@@@@@@@@ @@@@@@@@@@@@@+ :@@@@@@@@@@@ @@@@@@@@@@@@* :%@@@@@@@@@@@ @@@@@@@@@@@% -@@@@@@@@@@@@ @@@@@@@@@@@= %@@@@@@@@@@@@ @@@@@@@@@@#. -@@@@@@@@@@@@@ @@@@@@@@@@ :@@@@@@@@@@@@@@ @@@@@@@@@@ *@@@@@@@@@@@@@@ @@@@@@@@@- %@@@@@@@@@@@@@@ @@@@@@@@#. =@@@@@@@@@@@@@@@ @@@@@@@@# .@@@@@@@@@@@@@@@@ @@@@@@@@# .@@@@@@@@@@@@@@@@ @@@@@@@@%: .@@@@@@@@@@@@@@@@ @@@@@@@@@@@@@@@@@@@@@@@@@@@@ @@@@@@@@@@@@@@@@@@@@@@@@@@@@ @@@@@@@@@@@@@@@@@@@@@@@@@@@@ [01/10/2021-20:19:42] [I] Output: 0: 1: ********** 2: 3: 4: 5: 6: 7: 8: 9: &&&& PASSED TensorRT.sample_mnist # ./sample_mnist ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected] --------------------------------------------------------------------- To unsubscribe, e-mail: [email protected] For additional commands, e-mail: [email protected]
