environment.yml

name: immunogold
channels:
  - pytorch
  - nvidia
  - conda-forge
  - defaults
dependencies:
  - python=3.10
  - pytorch>=2.0
  - torchvision>=0.15
  - torchaudio
  - pytorch-cuda=11.8
  - numpy>=1.24
  - scipy>=1.10
  - pandas>=2.0
  - scikit-image>=0.21
  - matplotlib>=3.7
  - tifffile>=2023.4
  - pyyaml>=6.0
  - pip
  - pip:
    - albumentations>=1.3
    - opencv-python-headless>=4.7
    - tensorboard>=2.13
    - wandb
    - tqdm