From d055f97457918b1dde229650fd36d28bc50c40b2 Mon Sep 17 00:00:00 2001
From: mrava87 <matteoravasi@gmail.com>
Date: Fri, 21 Jun 2024 19:03:57 +0300
Subject: [PATCH] build: force torch cpuonly

---
 Makefile                | 2 +-
 environment-dev-arm.yml | 1 +
 environment-dev.yml     | 1 +
 requirements-dev.txt    | 1 +
 4 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/Makefile b/Makefile
index 341c53be..35cf0e1e 100755
--- a/Makefile
+++ b/Makefile
@@ -1,7 +1,7 @@
 PIP := $(shell command -v pip3 2> /dev/null || command which pip 2> /dev/null)
 PYTHON := $(shell command -v python3 2> /dev/null || command which python 2> /dev/null)
 
-.PHONY: install dev-install install_conda dev-install_conda tests doc docupdate
+.PHONY: install dev-install install_conda dev-install_conda tests doc docupdate servedoc lint typeannot coverage
 
 pipcheck:
 ifndef PIP
diff --git a/environment-dev-arm.yml b/environment-dev-arm.yml
index 7cb73753..4e666aab 100755
--- a/environment-dev-arm.yml
+++ b/environment-dev-arm.yml
@@ -10,6 +10,7 @@ dependencies:
   - numpy>=1.21.0
   - scipy>=1.4.0
   - pytorch>=1.2.0
+  - cpuonly
   - pyfftw
   - pywavelets
   - sympy
diff --git a/environment-dev.yml b/environment-dev.yml
index 59b2c127..f7d01a82 100755
--- a/environment-dev.yml
+++ b/environment-dev.yml
@@ -10,6 +10,7 @@ dependencies:
   - numpy>=1.21.0
   - scipy>=1.4.0
   - pytorch>=1.2.0
+  - cpuonly
   - pyfftw
   - pywavelets
   - sympy
diff --git a/requirements-dev.txt b/requirements-dev.txt
index d86f07f1..35d7d07e 100644
--- a/requirements-dev.txt
+++ b/requirements-dev.txt
@@ -1,5 +1,6 @@
 numpy>=1.21.0
 scipy>=1.4.0
+--extra-index-url https://download.pytorch.org/whl/cpu
 torch>=1.2.0
 numba
 pyfftw