diff --git a/patch/npu.patch b/patch/npu.patch
index c54cd60ab31caaa3aad713e1020a0332d7ec46b1..6271d928837bf7c5281f38a1def3814bba826a32 100644
--- a/patch/npu.patch
+++ b/patch/npu.patch
@@ -2642,14 +2642,14 @@ diff -Nur '--exclude=.git' apex/setup.py apex-develop/setup.py
 +secure_link_args = ['-Wl,-z,now', '-Wl,-z,relro', '-Wl,-z,noexecstack', '-s']
 +
 +def get_package_dir():
-+    if '--user' in sys.argv:
-+        package_dir = site.getusersitepackages()
-+    else:
-+        py_version = f'{sys.version_info.major}.{sys.version_info.minor}'
-+        package_dir = f'{sys.prefix}/lib/python{py_version}/site-packages'
++    torch_file_path = torch.__file__
++    package_dir = os.path.dirname(os.path.dirname(torch_file_path))
 +    return package_dir
 +
 +
++
++
++
 +def CppExtension(name, sources, *args, **kwargs):
 +    r'''
 +    Creates a :class:`setuptools.Extension` for C++.