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