chore(ml):ライブラリを修正

Doer-org · Nov 1, 2023 · 519100c · 519100c
1 parent 9d11390
commit 519100c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ml/pyproject.toml b/ml/pyproject.toml
@@ -13,7 +13,8 @@ omegaconf = "^2.3.0"
 pillow = "^10.0.0"
 sympy = "^1.12"
 torch = [
-    { version = "^2.0.1+cu118", source = "torch_cu118", markers = "sys_platform != 'darwin'" },
+    { url = "https://download.pytorch.org/whl/cu118/torch-2.0.1%2Bcu118-cp311-cp311-win_amd64.whl", markers = "sys_platform == 'win32'" },
+    { version = "^2.0.1+cu118", source = "torch_cu118", markers = "sys_platform == 'linux'" },
     { version = "^2.0.1+cpu", source = "torch_cpu", markers = "sys_platform == 'darwin'" },
 ]
 python-dotenv = "^1.0.0"