torch>=1.8.1

[dev]
cudatoolkit
pytorch
torchaudio
torchvision
matplotlib
tqdm
pytest
pytest-cov
black
flake8
flake8-bugbear
mypy
usort

[test]
cudatoolkit
pytorch
torchaudio
torchvision
matplotlib
tqdm
pytest
pytest-cov
