[GitHub] [incubator-mxnet] TristonC edited a comment on issue #17855: TensorRT does not work on other GPUs except GPU0

TristonC edited a comment on issue #17855: TensorRT does not work on other GPUs except GPU0 URL: https://github.com/apache/incubator-mxnet/issues/17855#issuecomment-604091749 @Caenorst Looks like the TRTCreateState function in

[GitHub] [incubator-mxnet] TristonC edited a comment on issue #17855: TensorRT does not work on other GPUs except GPU0

TristonC edited a comment on issue #17855: TensorRT does not work on other GPUs except GPU0 URL: https://github.com/apache/incubator-mxnet/issues/17855#issuecomment-604039572 It seems the onnx-trt not setting the device properly. @SayHiRay If you accept a dirty workaround try replace ctx

[GitHub] [incubator-mxnet] TristonC edited a comment on issue #17855: TensorRT does not work on other GPUs except GPU0

TristonC edited a comment on issue #17855: TensorRT does not work on other GPUs except GPU0 URL: https://github.com/apache/incubator-mxnet/issues/17855#issuecomment-604039572 It seems the onnx-trt not setting the device properly. If you accept a dirty workaround try replace ctx =

[GitHub] [incubator-mxnet] TristonC edited a comment on issue #17855: TensorRT does not work on other GPUs except GPU0

TristonC edited a comment on issue #17855: TensorRT does not work on other GPUs except GPU0 URL: https://github.com/apache/incubator-mxnet/issues/17855#issuecomment-604039572 It seems the TRT setting the device properly. If you accept a dirty workaround try replace ctx = mx.gpu(1) with