# definition file to build pytorch-caney conda environment
name: pytorch-caney

channels:
  - conda-forge
  - rapidsai
  - nvidia

dependencies:
  - python=3.9
  - cudatoolkit=11.8
  - pip
  - gdal
  - pip:
    - torch>=2.0.0
    - torchvision>=0.15
    - pytorch-lightning
    - omegaconf
    - rasterio
    - rioxarray
    - xarray
    - geopandas
    - opencv-python
    - opencv-python-headless
    - opencv-contrib-python
    - opencv-contrib-python-headless
    - tifffile
    - webcolors
    - Pillow
    - seaborn
    - xgboost
    - tiler
    - segmentation-models
    - pytest
    - coveralls
    - rtree
    - sphinx
    - sphinx_rtd_theme
    - yacs
    - termcolor
    - numba
    - joblib
    - segmentation-models-pytorch