From b1626e940ed7d7ce3bc1c3fcfb7370873f2ca602 Mon Sep 17 00:00:00 2001 From: Qiusheng Wu Date: Thu, 18 Jan 2024 00:22:48 -0500 Subject: [PATCH] Update GPU installation instructions (#239) * Update GPU installation instructions * Update jupyter server command --- Dockerfile | 2 +- README.md | 10 +++++++++- docs/installation.md | 6 ++++++ 3 files changed, 16 insertions(+), 2 deletions(-) diff --git a/Dockerfile b/Dockerfile index 0371ddd5..1d1ab99c 100644 --- a/Dockerfile +++ b/Dockerfile @@ -4,7 +4,7 @@ LABEL repo="https://github.com/opengeos/segment-geospatial" RUN mamba install -c conda-forge leafmap localtileserver segment-geospatial -y && \ pip install -U segment-geospatial jupyter-server-proxy && \ - jupyter serverextension enable --sys-prefix jupyter_server_proxy && \ + jupyter server extension enable --sys-prefix jupyter_server_proxy && \ fix-permissions "${CONDA_DIR}" && \ fix-permissions "/home/${NB_USER}" diff --git a/README.md b/README.md index 146779e1..2c4c6fcd 100644 --- a/README.md +++ b/README.md @@ -56,11 +56,19 @@ conda install -c conda-forge mamba mamba install -c conda-forge segment-geospatial ``` +If your system has a GPU, but the above commands do not install the GPU version of pytorch, you can force the installation of the GPU version of pytorch using the following command: + +```bash +mamba install -c conda-forge segment-geospatial "pytorch=*=cuda*" +``` + +````bash + Samgeo-geospatial has some optional dependencies that are not included in the default conda environment. To install these dependencies, run the following command: ```bash mamba install -c conda-forge groundingdino-py segment-anything-fast -``` +```` As of July 9th, 2023 Linux systems have also required that `libgl1` be installed for segment-geospatial to work. The following command will install that dependency diff --git a/docs/installation.md b/docs/installation.md index a02e9b63..c7337547 100644 --- a/docs/installation.md +++ b/docs/installation.md @@ -20,6 +20,12 @@ conda install -c conda-forge mamba mamba install -c conda-forge segment-geospatial ``` +If your system has a GPU, but the above commands do not install the GPU version of pytorch, you can force the installation of the GPU version of pytorch using the following command: + +```bash +mamba install -c conda-forge segment-geospatial "pytorch=*=cuda*" +``` + Samgeo-geospatial has some optional dependencies that are not included in the default conda environment. To install these dependencies, run the following command: ```bash