Install different packages depending on platform

Don't install cuda packages on Macs.
reginabarzilaygroup · Nov 29, 2023 · c2c17ca · c2c17ca
1 parent b0cf1dc
commit c2c17ca
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -36,8 +36,10 @@ python_requires = >=3.8
 install_requires =
     importlib-metadata; python_version>="3.8"
     numpy==1.24.1
-    torch==1.11.0+cu113
-    torchvision==0.12.0+cu113
+    torch==1.11.0+cu113; sys_platform != "darwin"
+    torch==1.11.0; sys_platform == "darwin"
+    torchvision==0.12.0+cu113; sys_platform != "darwin"
+    torchvision==0.12.0; sys_platform == "darwin"
     pytorch_lightning==1.5.6
     scikit-learn==1.0.2
     tqdm==4.62.3