fix typo

Signed-off-by: Jinzhe Zeng <[email protected]>
njzjz · Jun 21, 2024 · 30b5377 · 30b5377
1 parent ced2562
commit 30b5377
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepmd/tf/lmp.py b/deepmd/tf/lmp.py
@@ -80,7 +80,7 @@ def get_library_path(module: str, filename: str) -> List[str]:
 
 tf_dir = tf.sysconfig.get_lib()
 op_dir = str(SHARED_LIB_DIR)
-pt_spec = find_spec("torch")
+pt_spec = find_spec("torch.lib")
 if pt_spec is not None:
     pt_dir = pt_spec.submodule_search_locations[0]
 else: