Include missing import in TensorRT tutorial (#12609)

apache · Sep 21, 2018 · cb6815c · cb6815c
1 parent 1045140
commit cb6815c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/docs/tutorials/tensorrt/inference_with_trt.md b/docs/tutorials/tensorrt/inference_with_trt.md
@@ -29,6 +29,7 @@ TensorRT is an inference only library, so for the purposes of this blog post we
 import mxnet as mx
 from mxnet.gluon.model_zoo import vision
 import time
+import os
 
 batch_shape = (1, 3, 224, 224)
 resnet18 = vision.resnet18_v2(pretrained=True)