diff --git a/README.md b/README.md index b6e7ca59..bd53f506 100644 --- a/README.md +++ b/README.md @@ -100,7 +100,7 @@ conda install -c conda-forge groundingdino-py segment-anything-fast ### Install SAM 3 on Windows -It is a bit tricky to install SAM 3 on Windows. Run the following commands on Windows to install SamGeo: +Run the following commands to install SamGeo using `conda`: ```bash conda create -n geo python=3.12 @@ -110,6 +110,13 @@ pip install "segment-geospatial[samgeo3]" pip install triton-windows ipykernel jupyterlab ``` +`uv` based installation +```bash +uv venv --python=3.16 +uv pip install torch torchvision --index-url https://download.pytorch.org/whl/cu128 +uv pip install segment-geospatial[samgeo3] +``` + ## Examples - [Segmenting remote sensing imagery](https://samgeo.gishub.org/examples/satellite) diff --git a/pyproject.toml b/pyproject.toml index 34d9af10..9cde2b40 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -76,10 +76,12 @@ samgeo3 = [ "buildingregulariser", "leafmap", "localtileserver", + "psutil; platform_system == 'Windows'", "spacy", - "triton>=3.5.1; sys_platform == 'linux'", - "triton-windows>=3.3.0.post19; sys_platform == 'windows'", - "transformers>=5.0.0rc0", + "tokenizers==0.22.1", + "triton; platform_system == 'Linux'", + "triton-windows; platform_system == 'Windows'", + "transformers", "xarray" ] text = [