We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 757cbf8 commit f6cf4fdCopy full SHA for f6cf4fd
trainer/requirements.txt
@@ -5,6 +5,12 @@ pytest==8.4.2
5
scikit-image==0.25.2
6
scipy==1.16.2
7
pillow==12.0.0
8
-torch==2.6.0+cu124
9
-torchvision==0.21.0+cu124
10
+# ---- PyTorch: platform-specific pins ----
+# 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