diff --git a/docs/user/installation.rst b/docs/user/installation.rst
index 6b23ba7d7f7..ea9007ddc3b 100644
--- a/docs/user/installation.rst
+++ b/docs/user/installation.rst
@@ -67,8 +67,7 @@ Conda does not directly support installing development versions, but you can use
    $ conda activate torchgeo
    $ pip install .
 
-
-.. note:: If you do not have access to a GPU or are running on macOS, replace ``pytorch-gpu`` with ``pytorch-cpu`` in the ``environment.yml`` file.
+.. note:: The above method will not work on Windows. Windows users are recommended to create a conda environment and install dependencies via pip.
 
 Conda does not directly support optional dependencies. If you install from conda-forge, only required dependencies will be installed by default. Optional dependencies can be installed afterwards using pip. If you install using the ``environment.yml`` file, all optional dependencies are installed by default.
 
diff --git a/environment.yml b/environment.yml
index bf7d75ef32d..c103271b80a 100644
--- a/environment.yml
+++ b/environment.yml
@@ -11,7 +11,7 @@ dependencies:
   - pycocotools
   - pyproj>=2.2
   - python>=3.6
-  - pytorch-gpu>=1.7
+  - pytorch>=1.7
   - rarfile>=3
   - rasterio>=1.0.16
   - shapely>=1.3