FluStudy/environment.yml at master · behavioral-data/FluStudy · GitHub

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
name: seattleflustudy
channels:
  - conda-forge
  - defaults
  - pytorch
dependencies:
  - python=3.9
  - pip
  - click=7.1
  - Sphinx
  - coverage=5.5
  - python-dotenv>=0.5.1
  - pyarrow=3.0.0
  - numpy=1.20
  - transformers=4.11.3
  - pandas=1.2
  - scikit-learn=0.24
  - scipy=1.6.0
  - dask=2021.02.0
  - distributed=2021.02.0
  - pylint=2.7.2
  - tensorflow-gpu
  - fastparquet
  - ipython
  - graphviz
  - pathlib2
  - pytorch
  - torchvision
  - torchaudio
  - cudatoolkit=10.2
  - jupyter
  - plotly
  - dask-ml
  - matplotlib
  - pytorch-lightning=1.4.9
  - torchmetrics>=0.5
  - tensorboardx
  - pyspark=3.1.2
  - pip:
    - torchviz==0.0.2
    - pyyaml==5.4.1
    - wandb
    - xgboost==1.4.2
    - tensorboard==2.6.0
    - ray[default]
    - tabulate==0.8.9
    - petastorm==0.11.3
    - methodtools==0.4.5
variables:
  USE_TORCH: 1