Makefile torch-cuda requirements process

GitGinocchio · GitGinocchio · commit 0bfc9fa57440 · 2025-02-10T23:16:42.000+01:00
diff --git a/Makefile b/Makefile
@@ -0,0 +1,6 @@
+
+CU_VERSION = cu124
+
+requirements:
+	pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/$(CU_VERSION)
+	pip install -r requirements.txt
diff --git a/requirements.txt b/requirements.txt
@@ -21,9 +21,9 @@ setuptools==75.8.0
 six==1.17.0
 sympy==1.13.1
 threadpoolctl==3.5.0
-torch==2.6.0                #+cu124
-torchaudio==2.6.0           #+cu124
-torchvision==0.21.0         #+cu124
+torch==2.6.0                #+cu124 If you want to install the cuda version see Makefile
+torchaudio==2.6.0           #+cu124 If you want to install the cuda version see Makefile
+torchvision==0.21.0         #+cu124 If you want to install the cuda version see Makefile
 tqdm==4.67.1
 typing_extensions==4.12.2
 tzdata==2025.1
