diff --git a/deepmd/pt/cxx_op.py b/deepmd/pt/cxx_op.py index 7c21c8c9cc..d46f20a0bc 100644 --- a/deepmd/pt/cxx_op.py +++ b/deepmd/pt/cxx_op.py @@ -50,10 +50,10 @@ def load_library(module_name: str) -> bool: if PT_CXX11_ABI_FLAG != pt_cxx11_abi_flag: raise RuntimeError( "This deepmd-kit package was compiled with " - "CXX11_ABI_FLAG=%d, but TensorFlow runtime was compiled " + "CXX11_ABI_FLAG=%d, but PyTorch runtime was compiled " "with CXX11_ABI_FLAG=%d. These two library ABIs are " "incompatible and thus an error is raised when loading %s. " - "You need to rebuild deepmd-kit against this TensorFlow " + "You need to rebuild deepmd-kit against this PyTorch " "runtime." % ( PT_CXX11_ABI_FLAG,