From bc5c5997176575444c597437a5976b180c4cfdbb Mon Sep 17 00:00:00 2001
From: Jacob Silterra <silterra@mit.edu>
Date: Fri, 31 May 2024 11:34:49 -0400
Subject: [PATCH] Change setup.cfg so that we install torch+cuda based on
 platform_machine (architecture) not OS.

---
 setup.cfg | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/setup.cfg b/setup.cfg
index 54db2c9..c76970a 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -46,11 +46,11 @@ install_requires =
     pydicom==2.3.0
     pylibjpeg[all]==2.0.0
     scikit-learn==1.0.2
-    torch==1.13.1+cu117; sys_platform != "darwin"
-    torch==1.13.1; sys_platform == "darwin"
+    torch==1.13.1+cu117; platform_machine == "x86_64"
+    torch==1.13.1; platform_machine != "x86_64"
     torchio==0.18.74
-    torchvision==0.14.1+cu117; sys_platform != "darwin"
-    torchvision==0.14.1; sys_platform == "darwin"
+    torchvision==0.14.1+cu117; platform_machine == "x86_64"
+    torchvision==0.14.1; platform_machine != "x86_64"
     tqdm==4.62.3
 
 [options.packages.find]