update

2023-05-17 21:20:41 +08:00
parent 11813454de
commit df8eef5f64
14 changed files with 438 additions and 121 deletions
--- a/backend-python/utils/torch.py
+++ b/backend-python/utils/torch.py
@@ -3,7 +3,7 @@ import sysconfig


 def set_torch():
-    torch_path = os.path.join(sysconfig.get_paths()["purelib"], "torch\\lib")
+    torch_path = os.path.join(sysconfig.get_paths()["purelib"], f"torch{os.sep}lib")
    paths = os.environ.get("PATH", "")
    if os.path.exists(torch_path):
        print(f"torch found: {torch_path}")