Add back pycuda.autoinit to test_pt_onnx_trt (pytorch#51106)

Summary:
Fixes pytorch#51105 by adding back the `import pycuda.autoinit`.

Pull Request resolved: pytorch#51106

Reviewed By: mingzhe09088

Differential Revision: D26086808

Pulled By: heitorschueroff

fbshipit-source-id: 88d98796c87a44cedaa1f6666e9f71a424293641

caffe2/python/trt/test_pt_onnx_trt.py

@@ -22,6 +22,8 @@
 import torchvision.models as models
 
 import pycuda.driver as cuda
+# This import causes pycuda to automatically manage CUDA context creation and cleanup.
+import pycuda.autoinit
 
 import tensorrt as trt
 TRT_LOGGER = trt.Logger(trt.Logger.WARNING)