REVA-QCAV/env.yml

36 lines
446 B
YAML
Raw Normal View History

2023-04-01 17:10:27 +00:00
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
2023-04-01 17:10:27 +00:00
# deep learning libraries
- transformers
- datasets
- timm
# dev tools
- ruff
2023-04-01 17:31:20 +00:00
- black
2023-04-01 17:10:27 +00:00
- isort
- mypy
- pre-commit
# logging
- tensorboard
# visualization
- matplotlib