Merge pull request #587 from mrava87/patch-torchcpu

build: force torch cpuonly

Author: mrava87

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

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

environment-dev.yml

@@ -10,6 +10,7 @@ dependencies:
   - numpy>=1.21.0
   - scipy>=1.4.0
   - pytorch>=1.2.0
+  - cpuonly
   - pyfftw
   - pywavelets
   - sympy

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