environment.yml

name: xt
channels: 
  - pytorch
  - nvidia
  - conda-forge
dependencies:
  - python=3.9
  - pytorch==2.0.1
  - torchvision==0.15.2
  - pytorch-cuda=11.8
  - scikit-image
  - rasterio==1.2.10
  - gdal
  - shapely
  - torchmetrics
  - pip
  - pip:
    - numpy>=1.21.0,<2.0.0
    - timm==0.9.7
    - wandb
    - pre-commit>=2.15.0,<3.0.0
    - black[jupyter]>=21.8,<23
    - flake8>=3.8,<5
    - isort[colors]>=5.8,<6
    - pydocstyle[toml]>=6.1,<7
    - pyupgrade>=1.24,<3
    - pytest>=6.1.2,<8
    - pytest-cov>=2.4,<4
    - types-requests>=2.28.9,<2.30.0
    - types-python-dateutil>=2.8.19,<2.9.0
    - kornia==0.6.7
    - torchgeo
    - classy-vision==0.6.0
    - tensorboard
    - causal-conv1d>=1.1.0,<1.2.0
    - mamba-ssm
    - opencv-python
    - einops
    - albumentations
    - tensorboardX
    - madgrad
    - warmup_scheduler
    - wandb-osh
    - submitit
    - tifffile
    - omegaconf
    - pycocotools
    - prettytable
    - triton==2.0.0.dev20221202
    - hiera-transformer