Cross-Resolution-Dead-Tree-Segmentation/env_cuda12.yml at master · YanCheng-go/Cross-Resolution-Dead-Tree-Segmentation · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
name: ssl4eo_delfors  # worked for deic hpc
channels:
  - pytorch
  - nvidia
  - conda-forge
  - defaults
  - balbasty
dependencies:
  - python>=3.9, <3.11
  - pytorch
  - pytorch-cuda=12.1
  - torchvision
  - tqdm
  - numpy<2.0
  - numba
  - einops
  - wandb
  - pytorch-lightning
  - lightning-bolts
  - torchmetrics
  - scipy
  - timm
  - matplotlib
  - pandas=1.4.1
  - lightning
  - umap-learn
  - h5py
  - cupy
  - pkg-config
  - libjpeg-turbo
  - conda-forge::opencv
  - ipykernel
  - jupyterlab
  - gdal=3.2.2
  - geopandas>=0.10.2
  - rasterio=1.2.6
  - shapely=1.8.0
  - proj=8.0.1
  - qt=5.12.9
  - pyproj=3.3.1
  - torch-distmap
  - tensorboard=2.8.0
  - tensorboardx=2.2
  - scikit-image
  - scienceplots
  - docstring_parser
  - seaborn
  - ipdb
  - pytest
  - lightly
  - black
  - contextily
  - pip
  - pip:
      # - geobench
      - ffcv
      - kornia
      # - segmentation-models-pytorch==0.3.4 # 0.3.4 dev. version, pip install -U git+https://github.com/qubvel/segmentation_models.pytorch
      - opencv-python
      - alembic
      - geoalchemy2
      - geopy
      - openpyxl
      - pylint
      - python-dateutil