cpu only tests

geoopt · May 23, 2020 · dd7bd12 · dd7bd12
1 parent e748ec0
commit dd7bd12
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/CHANGELOG.rst b/CHANGELOG.rst
@@ -7,6 +7,9 @@ New Features
 ------------
 * BirkhoffPolytope (#125)
 
+Maintenance
+-----------
+* Tests for pytorch>=1.4, cpuonly (#133)
 
 geoopt v0.1.2
 ==============

diff --git a/scripts/install_deps.sh b/scripts/install_deps.sh
@@ -1,3 +1,3 @@
 conda install --yes pip numpy mkl-service
-conda install --yes pytorch${PYTORCH_VERSION} -c ${PYTORCH_CHANNEL}
+conda install --yes pytorch${PYTORCH_VERSION} cpuonly -c ${PYTORCH_CHANNEL}
 pip install -r ./requirements-dev.txt