SBDM/environment.yml at main · ai-med/SBDM · 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
66
67
name: sbdm
channels:
  - pytorch
  - conda-forge
  - defaults
dependencies:
  # Python
  - python=3.10

  # Core ML libraries
  - pytorch>=2.0.0
  - torchvision>=0.15.0
  - torchaudio>=2.0.0
  - numpy>=1.21.0
  - scipy>=1.9.0
  - scikit-learn>=1.1.0

  # Data processing
  - pandas>=1.5.0
  - h5py>=3.7.0
  - pyarrow>=10.0.0

  # Visualization
  - matplotlib>=3.5.0
  - seaborn>=0.11.0
  - plotly>=5.0.0

  # Medical imaging
  - nibabel>=5.0.0
  - trimesh>=3.15.0
  - vtk>=9.0.0
  - pyvista>=0.40.0

  # Utilities
  - tqdm>=4.64.0
  - requests>=2.28.0
  - pyyaml>=6.0.0
  - pillow>=9.0.0
  - psutil>=5.9.0

  # Development tools
  - jupyter>=1.0.0
  - ipykernel>=6.0.0

  # Pip dependencies
  - pip
  - pip:
    # Configuration management
    - hydra-core>=1.3.0
    - omegaconf>=2.3.0

    # Deep learning utilities
    - accelerate>=0.20.0
    - ema-pytorch>=0.6.0
    - einops>=0.6.0
    - bidict>=0.22.0

    # Logging and monitoring
    - wandb>=0.13.0

    # Development and testing
    - pytest>=7.0.0
    - pytest-cov>=4.0.0
    - black>=22.0.0
    - isort>=5.10.0
    - flake8>=5.0.0
    - mypy>=1.0.0