ProdSegmenter/environment-gpu.yml at main · nerdy1texan/ProdSegmenter · 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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
name: prodsegmenter
channels:
  - defaults
  - conda-forge
  - pytorch
  - nvidia
dependencies:
  # Python and core
  - python=3.10
  - pip

  # Jupyter and development
  - jupyterlab
  - jupyter
  - ipykernel

  # Core ML and scientific computing
  - numpy>=1.21.0
  - scipy
  - scikit-learn
  - pandas

  # Computer vision and image processing
  - opencv
  - pillow
  - matplotlib
  - seaborn

  # Deep learning - PyTorch with CUDA 12.1 (RTX 3060 compatible)
  - pytorch>=2.0.0
  - torchvision
  - torchaudio
  - pytorch-cuda=12.1
  - cudatoolkit

  # Additional ML libraries
  - scikit-image

  # Development and testing
  - pytest
  - pytest-cov
  - black
  - flake8
  - pre-commit

  # Utilities
  - tqdm
  - click
  - pyyaml
  - requests

  # Pip dependencies
  - pip:
    # Segment Anything Model (GPU accelerated)
    - segment-anything

    # Azure dependencies
    - azureml-sdk[notebooks,automl]
    - azureml-mlflow
    - azure-storage-blob
    - azure-identity
    - azure-keyvault-secrets

    # MLOps and experiment tracking
    - mlflow>=2.0.0
    - wandb

    # Web framework
    - streamlit>=1.25.0
    - fastapi
    - uvicorn

    # Image augmentation (GPU accelerated)
    - albumentations>=1.3.0

    # Video processing
    - ffmpeg-python
    - imageio[ffmpeg]
    - moviepy

    # Additional ML utilities
    - timm
    - transformers
    - accelerate

    # Monitoring and logging
    - tensorboard
    - rich

    # Performance
    - numba

    # Configuration management
    - hydra-core
    - omegaconf

    # GPU-accelerated image processing
    - cupy-cuda12x