diff --git a/Dockerfile b/Dockerfile
new file mode 100644
index 0000000..8f2705d
--- /dev/null
+++ b/Dockerfile
@@ -0,0 +1,9 @@
+ARG TENSORRT="7"
+ARG CUDA="10"
+
+FROM hakuyyf/tensorrtx:trt${TENSORRT}_cuda${CUDA}
+
+RUN apt-get update
+
+# git clone tensorrtx
+RUN git clone https://github.com/wang-xinyu/tensorrtx.git
\ No newline at end of file