projet-long/environment.yml
Laurent Fainsin d8bba4481a refactor!: move files to python modules
feat: add back (broken) faster rcnn model
2023-04-05 20:11:01 +02:00

36 lines
487 B
YAML

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