-
Notifications
You must be signed in to change notification settings - Fork 0
/
environment.yaml
46 lines (44 loc) · 759 Bytes
/
environment.yaml
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
name: plankton
channels:
- nvidia
- pytorch
- conda-forge
dependencies:
- python=3.9
- pip
- cudatoolkit=11.3
- pytorch>=1.11
- natsort
- pre-commit
- black
- isort
- flake8
- tensorboard
- torchvision>=0.12
- pytorch-lightning>=1.6
- torchmetrics
- matplotlib
- ipython
- hydra-core=1.1
- jupyterlab
- scikit-learn
- rich
- pandas
- dask
- distributed
- numpy
- seaborn
- python-dotenv
- onnx
- pillow
- tqdm
- pip:
- onnxruntime-gpu
- lightning-bolts
- torch-tb-profiler
- hydra-colorlog
- hydra-optuna-sweeper
- hydra-submitit-launcher
- test-tube
- catalyst
- git+https://github.com/ufoym/imbalanced-dataset-sampler.git # torchsampler