add Dockerfile (#291)

wang-xinyu · Nov 17, 2020 · 659fd2b · 659fd2b
1 parent 6adbe35
commit 659fd2b
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/Dockerfile 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