diff --git a/docker/WORKSPACE.ngc b/docker/WORKSPACE.ngc index 3e570c63f1..a01bbed32b 100755 --- 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 index 72483786eb..2629c0391b 100644 --- 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"] )