Skip to content

Commit 4807d8d

Browse files
committed
chore: Add trtorch external dependency option in the dockerfile
Signed-off-by: Dheeraj Peri <[email protected]>
1 parent b5c324a commit 4807d8d

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

docker/WORKSPACE.cu.docker

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,13 @@ git_repository(
2929
shallow_since = "1570114335 -0400"
3030
)
3131

32+
# External dependency for trtorch if you already have precompiled binaries.
33+
# This is currently used in pytorch NGC container CI testing.
34+
local_repository(
35+
name = "trtorch",
36+
path = "/opt/pytorch/trtorch"
37+
)
38+
3239
# CUDA should be installed on the system locally
3340
new_local_repository(
3441
name = "cuda",

0 commit comments

Comments
 (0)