name: pytorch_gpu
channels:
  - pytorch
  - nvidia
  - conda-forge
dependencies:
  - python=3.10
  - pytorch::pytorch
  - pytorch::torchvision
  - pytorch::torchaudio
  - pytorch::pytorch-cuda=11.8
  - gdal>=3.7.0
  - pip
  - pip:
    - sentinelhub>=3.9.0
    - shapely>=2.0.0
    - pyproj>=3.4.0