Put torch in requirements-test.txt [skip ci]

andreajparker · Feb 12, 2019 · 60a2319 · 60a2319
1 parent 676fddc
commit 60a2319
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/README.rst b/README.rst
@@ -43,9 +43,9 @@ dependencies and running the tests and linter.
 Installing dependencies
 -----------------------
 
-Make sure you setup a virtual environment with Python and PyTorch
-installed. Then, install all the dependencies in ``requirements.txt`` file and
-install this package in development mode.
+Make sure you setup a virtual environment with Python. Then, install all
+the dependencies in ``requirements.txt`` file and install this package in
+development mode.
 
 ::
 

diff --git a/requirements-test.txt b/requirements-test.txt
@@ -1,2 +1,6 @@
+# This only installs PyTorch with a specific CUDA version which may not be
+# compatible with yours. If so, install PyTorch with the correct CUDA version
+# as instructed on https://pytorch.org/get-started/locally/
+torch
 pytest==3.2.5
 pytest-cov==2.5.1