name: qcav

channels:
  - nodefaults
  - pytorch
  - nvidia
  - conda-forge

dependencies:
  # basic python
  - rich
  # science
  - numpy
  - scipy
  - opencv
  # pytorch
  - pytorch
  - torchvision
  - torchaudio
  - pytorch-cuda
  - lightning # currently broken, install manually with pip
  # deep learning libraries
  - transformers
  - datasets
  - timm
  # dev tools
  - ruff
  - black
  - isort
  - mypy
  - pre-commit
  # logging
  - tensorboard
  # visualization
  - matplotlib