-
Notifications
You must be signed in to change notification settings - Fork 15
/
environment.yml
54 lines (52 loc) · 1.18 KB
/
environment.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
# Chnage project name. We recommend you to match this name to the name of the GitHub repository
name: AYolov2
channels:
- conda-forge
- defaults
- pytorch
dependencies:
- python=3.7.7 # Change python version if necessary
- pip
- black=22.3.0
- coverage=5.3
- flake8=3.8.3
- flake8-bugbear=20.1.4
- flake8-docstrings=1.5.0
- isort=5.5.3
- mypy=0.782
- pre-commit=2.7.1
- pyflakes=2.2.0
- pylint=2.6.0
- pytest=6.0.2
- pytest-cov=2.10.1
- pytest-flake8=1.0.6
- pytest-mypy=0.7.0
- pytest-pylint=0.17.0
# Place your conda dependencies here
- pytorch=1.9.1
- torchvision=0.10.1
- cudatoolkit=11.1
- opencv=4.1.0
- pip:
- flake8-annotations==2.4.0
- flake8-polyfill==1.0.2
- docformatter==1.4.0
- mkdocs==1.1.2
- mkapi==1.0.14
- mkdocs-ivory==0.4.6
- pytorch-lightning==1.4.9
# Place your pip dependencies here
- p-tqdm==1.3.3
- albumentations==1.1.0
- kindle>=0.4.12
- matplotlib==3.3.4
- seaborn==0.11.2
- onnx==1.6.0
- onnx-simplifier>=0.3.6
- onnxruntime==1.9.0
- einops==0.3.2
- wandb==0.12.3
- tensorly==0.6.0
- orjson==3.6.4
- optuna==2.10.0
- pycocotools==2.0.2