Upload code/envs/seggen.yml with huggingface_hub
Browse files- code/envs/seggen.yml +24 -0
code/envs/seggen.yml
ADDED
|
@@ -0,0 +1,24 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Main framework env: SMP / Attention-UNet / TransUNet / Swin-Unet + SegGuidedDiff
|
| 2 |
+
# Modern stack (matches detected host: CUDA 12.8). bf16 on A100, fp16 on V100.
|
| 3 |
+
name: seggen
|
| 4 |
+
channels: [conda-forge]
|
| 5 |
+
dependencies:
|
| 6 |
+
- python=3.11
|
| 7 |
+
- pip
|
| 8 |
+
- pip:
|
| 9 |
+
# install torch matching your CUDA (host has cu128):
|
| 10 |
+
# pip install torch torchvision --index-url https://download.pytorch.org/whl/cu128
|
| 11 |
+
- segmentation-models-pytorch
|
| 12 |
+
- albumentations==2.0.8 # pin: avoids AGPL AlbumentationsX on -U
|
| 13 |
+
- monai # transforms + HD95 metric
|
| 14 |
+
- medpy # HD95 fallback
|
| 15 |
+
- opencv-python-headless
|
| 16 |
+
- numpy
|
| 17 |
+
- pyyaml
|
| 18 |
+
- timm
|
| 19 |
+
- einops
|
| 20 |
+
- ml-collections # required by TransUNet config
|
| 21 |
+
- tqdm
|
| 22 |
+
# SegGuidedDiff (runs in this env too — modern deps):
|
| 23 |
+
- diffusers==0.21.4
|
| 24 |
+
- datasets==2.14.5
|