DLFW changes (#1878)

pytorch · May 30, 2023 · 838d26b · 838d26b
1 parent e73c482
commit 838d26b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/docker/WORKSPACE.ngc b/docker/WORKSPACE.ngc
@@ -33,7 +33,7 @@ git_repository(
 # This is currently used in pytorch NGC container CI testing.
 local_repository(
     name = "torch_tensorrt",
-    path = "/opt/python3/site-packages/torch_tensorrt/"
+    path = "/usr/local/lib/python3.10/dist-packages/torch_tensorrt/"
 )
 
 # CUDA should be installed on the system locally
@@ -55,13 +55,13 @@ new_local_repository(
 
 new_local_repository(
     name = "libtorch",
-    path = "/opt/python3/site-packages/torch/",
+    path = "/usr/local/lib/python3.10/dist-packages/torch/",
     build_file = "third_party/libtorch/BUILD"
 )
 
 new_local_repository(
     name = "libtorch_pre_cxx11_abi",
-    path = "/opt/python3/site-packages/torch/",
+    path = "/usr/local/lib/python3.10/dist-packages/torch/",
     build_file = "third_party/libtorch/BUILD"
 )
 

diff --git a/noxfile.py b/noxfile.py
@@ -5,7 +5,7 @@
 
 # Use system installed Python packages
 PYT_PATH = (
-    "/usr/local/lib/python3.8/dist-packages"
+    "/usr/local/lib/python3.10/dist-packages"
     if not "PYT_PATH" in os.environ
     else os.environ["PYT_PATH"]
 )