🧑‍💻 Install torch GPU by default

shaneahmed · shaneahmed · commit bcb79600cb9f · 2025-12-18T09:54:42.000Z
diff --git a/.readthedocs.yml b/.readthedocs.yml
@@ -22,5 +22,6 @@ sphinx:
 python:
   install:
       # Install PyTorch CPU version first
+      - requirements: requirements/requirements_cpu.txt
       - requirements: docs/requirements.txt
       - requirements: requirements/requirements.txt
diff --git a/docs/requirements.txt b/docs/requirements.txt
@@ -1,10 +1,6 @@
-
---index-url https://download.pytorch.org/whl/cpu
 furo>=2022.12.7
 myst-nb>=0.17.1
 myst-parser>=0.18.1
 sphinx-copybutton>=0.5.1
 sphinx-design>=0.3.0
 sphinx-toolbox>=3.2.0
-torch
-torchvision
diff --git a/requirements/requirements.txt b/requirements/requirements.txt
@@ -1,5 +1,3 @@
-# torch installation
---index-url https://download.pytorch.org/whl/cu126; sys_platform != "darwin"
 aiohttp>=3.8.1
 albumentations>=1.3.0
 bokeh>=3.1.1, <3.6.0
diff --git a/requirements/requirements_cpu.txt b/requirements/requirements_cpu.txt
@@ -0,0 +1,3 @@
+--index-url https://download.pytorch.org/whl/cpu
+torch
+torchvision

Original file line number	Diff line number	Diff line change
`@@ -1,5 +1,3 @@`
`1`		`-# torch installation`
`2`		`---index-url https://download.pytorch.org/whl/cu126; sys_platform != "darwin"`
`3`	`1`	`aiohttp>=3.8.1`
`4`	`2`	`albumentations>=1.3.0`
`5`	`3`	`bokeh>=3.1.1, <3.6.0`
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+--index-url https://download.pytorch.org/whl/cpu`
	`2`	`+torch`
	`3`	`+torchvision`