Update torchxla2 installation for torch on mac and others (#7990)

pytorch · Sep 10, 2024 · 6ff03e6 · 6ff03e6
1 parent 5c4bac5
commit 6ff03e6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/experimental/torch_xla2/dev-requirements.txt b/experimental/torch_xla2/dev-requirements.txt
@@ -1,3 +1,4 @@
 -f https://download.pytorch.org/whl/torch
-torch==2.4.0+cpu
+torch==2.4.0; sys_platform == 'darwin'  # macOS
+torch==2.4.0+cpu; sys_platform != 'darwin' # Non-macOS (CPU-only), like on TPU
 ruff~=0.3.5