$ python3
Python 3.6.9 (default, Oct 8 2020, 12:12:24)
[GCC 8.4.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import onnxruntime as rt
>>> sess = rt.InferenceSession("/usr/src/tensorrt/data/mnist/mnist.onnx")
>>> sess.set_providers(['CUDAExecutionProvider'])