REVA-QCAV/env.yml

36 lines
446 B
YAML

name: qcav
channels:
- nodefaults
- pytorch
- nvidia
- conda-forge
dependencies:
# basic python
- rich
# science
- numpy
- scipy
- opencv
# pytorch
- pytorch
- torchvision
- torchaudio
- pytorch-cuda=11.8
- lightning
# deep learning libraries
- transformers
- datasets
- timm
# dev tools
- ruff
- black
- isort
- mypy
- pre-commit
# logging
- tensorboard
# visualization
- matplotlib