chore: Add trtorch external dependency option in the dockerfile

Signed-off-by: Dheeraj Peri <peri.dheeraj@gmail.com>
pytorch · Oct 15, 2021 · 4807d8d · 4807d8d
1 parent b5c324a
commit 4807d8d
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/docker/WORKSPACE.cu.docker b/docker/WORKSPACE.cu.docker
@@ -29,6 +29,13 @@ git_repository(
     shallow_since = "1570114335 -0400"
 )
 
+# External dependency for trtorch if you already have precompiled binaries.
+# This is currently used in pytorch NGC container CI testing.
+local_repository(
+    name = "trtorch",
+    path = "/opt/pytorch/trtorch"
+)
+
 # CUDA should be installed on the system locally
 new_local_repository(
     name = "cuda",