torchgeo

环境要求

• 已安装 Python 与 pip 的本地环境
• 可选：uv 包管理器（README 提供的另一种安装方式）
• 可选：运行示例代码需额外安装 PyTorch 生态组件（示例中导入了 lightning.pytorch 与 torch）
• conda / spack 用户请查阅官方安装文档

安装与启动步骤

1. 1准备 Python 环境

   确认本机已有可用的 Python 和 pip；conda、spack 用户请先阅读官方安装文档再动手。

   python3 --version
   pip --version

   复制

2. 2用 pip 安装 TorchGeo

   README 推荐的安装方式，一条命令即可把 torchgeo 及其依赖装进当前环境。

   pip install torchgeo

   复制

3. 3或用 uv 安装

   如果习惯用 uv 管理依赖，README 给出的等价方式是 uv add，注意需在项目目录中执行。

   uv add torchgeo

   复制

4. 4运行命令行工具

   安装完成后可直接调用 torchgeo 命令；在源码目录中也可用模块方式启动，均支持命令行配置。

   torchgeo --help
   python3 -m torchgeo --help

   复制

5. 5跑通示例导入

   新建脚本或用交互式解释器执行 README 的导入片段，能无报错导入即说明库可用。

   from lightning.pytorch import Trainer
   from torch.utils.data import DataLoader
   
   from torchgeo.datamodules import InriaAerialImageLabelingDataModule
   from torchgeo.datasets import CDL, Landsat7, Landsat8, VHR10, stack_samples
   from torchgeo.samplers import RandomPatchSampler
   from torchgeo.tasks import SemanticSegmentation

   复制

如何确认成功

命令行执行 torchgeo（或 python3 -m torchgeo）能正常输出帮助信息，Python 中导入 torchgeo 各子模块不报错。

常见问题

注意事项

• README 只给出了 pip 与 uv 两种安装方式，conda/spack 请以官方文档为准。
• 命令行工具支持命令行参数或 YAML/JSON 配置文件两种配置途径。
• 示例代码依赖 lightning.pytorch 和 PyTorch，需自行确保这些依赖已安装。
• 更多 API 文档与教程见 https://torchgeo.org 与 ReadTheDocs。