DLFW changes (#1878)

Author: apbose

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"
 )
 

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"]
 )