From 659fd2b23482197b19dccf746a5a3dbff1611381 Mon Sep 17 00:00:00 2001
From: Alfeng Yuan <42317483+AlfengYuan@users.noreply.github.com>
Date: Tue, 17 Nov 2020 14:32:56 +0800
Subject: [PATCH] add Dockerfile (#291)

---
 Dockerfile | 9 +++++++++
 1 file changed, 9 insertions(+)
 create mode 100644 Dockerfile

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