From 7a81d661755786cbdda5663f7b7c3342d29366d3 Mon Sep 17 00:00:00 2001 From: wolearyc Date: Thu, 12 Sep 2024 15:16:20 -0700 Subject: [PATCH] specify platform --- torch_requirements.txt | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/torch_requirements.txt b/torch_requirements.txt index 59fbd9a..5a5ba1f 100644 --- a/torch_requirements.txt +++ b/torch_requirements.txt @@ -1 +1,3 @@ -torch==2.4.0 +torch==2.4.1;platform_system=="darwin" +torch==2.4.1+cpu;platform_system=="linux" +torch==2.4.1+cpu;platform_system=="Windows"