From 6b9845cc001a56fff9c59ddc558526fbeee46fd0 Mon Sep 17 00:00:00 2001
From: Ashwin Nair <ash1995@gmail.com>
Date: Fri, 17 Dec 2021 02:46:59 +0400
Subject: [PATCH] Replace pytorch-gpu with pytorch in environment.yml (#295)

* Replace pytorch-gpu with pytorch in environment.yml

* Add windows note
---
 docs/user/installation.rst | 3 +--
 environment.yml            | 2 +-
 2 files changed, 2 insertions(+), 3 deletions(-)

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