diff --git a/docker/WORKSPACE.cu.docker b/docker/WORKSPACE.cu.docker index c2ed3b9228..34fec5c983 100755 --- 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",