remove the error message when PT is not preinstalled

Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
njzjz · Jun 20, 2024 · 60b6903 · 60b6903
1 parent 32831df
commit 60b6903
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/backend/read_env.py b/backend/read_env.py
@@ -104,8 +104,6 @@ def get_argument_from_env() -> Tuple[str, list, list, dict, str]:
 
     if os.environ.get("DP_ENABLE_PYTORCH", "1") == "1":
         pt_install_dir, _ = find_pytorch()
-        if pt_install_dir is None:
-            raise RuntimeError("Cannot find installed PyTorch.")
         cmake_args.extend(
             [
                 "-DENABLE_PYTORCH=ON",