36 lines
487 B
YAML
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
|