Skip to content

Commit f6cf4fd

Browse files
committed
add conditional requirements for pytorch on darwin
1 parent 757cbf8 commit f6cf4fd

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

trainer/requirements.txt

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,12 @@ pytest==8.4.2
55
scikit-image==0.25.2
66
scipy==1.16.2
77
pillow==12.0.0
8-
torch==2.6.0+cu124
9-
torchvision==0.21.0+cu124
108

9+
# ---- PyTorch: platform-specific pins ----
10+
# macOS (MPS/CPU wheels from PyPI)
11+
torch==2.6.0; sys_platform == "darwin"
12+
torchvision==0.21.0; sys_platform == "darwin"
13+
14+
# Windows/Linux CUDA 12.4
15+
torch==2.6.0+cu124; sys_platform != "darwin"
16+
torchvision==0.21.0+cu124; sys_platform != "darwin"

0 commit comments

Comments
 (0)