Skip to content

Commit 2ed9280

Browse files
authored
Merge pull request #1435 from pytorch/torch_paths
chore: Update torch installation paths for NGC
2 parents 5a7f00e + 7634a64 commit 2ed9280

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

docker/WORKSPACE.ngc

+3-3
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ git_repository(
3333
# This is currently used in pytorch NGC container CI testing.
3434
local_repository(
3535
name = "torch_tensorrt",
36-
path = "/opt/conda/lib/python3.8/site-packages/torch_tensorrt"
36+
path = "/usr/local/lib/python3.8/dist-packages/torch_tensorrt"
3737
)
3838

3939
# CUDA should be installed on the system locally
@@ -55,13 +55,13 @@ new_local_repository(
5555

5656
new_local_repository(
5757
name = "libtorch",
58-
path = "/opt/conda/lib/python3.8/site-packages/torch",
58+
path = "/usr/local/lib/python3.8/dist-packages/torch",
5959
build_file = "third_party/libtorch/BUILD"
6060
)
6161

6262
new_local_repository(
6363
name = "libtorch_pre_cxx11_abi",
64-
path = "/opt/conda/lib/python3.8/site-packages/torch",
64+
path = "/usr/local/lib/python3.8/dist-packages/torch",
6565
build_file = "third_party/libtorch/BUILD"
6666
)
6767

0 commit comments

Comments
 (0)