From d055f97457918b1dde229650fd36d28bc50c40b2 Mon Sep 17 00:00:00 2001 From: mrava87 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