graphorge/environment.yml at main · bessagroup/graphorge · 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
name: graphorge_env
channels:
  - defaults
  - conda-forge
  - nvidia
  - pytorch
  - pyg
dependencies:
  - python=3.11.9
  - pip=23.3
  - cycler=0.11.0
  - matplotlib=3.8.0
  - numpy=1.26.4
  - pytest=7.1.2
  - coverage=7.3.2
  - scipy=1.11.4
  - sphinx_rtd_theme=1.2.0
  - tqdm=4.65.0
  - hydra-core=1.3.2
  - torchinfo=1.8.0
  # Check:
  # https://pytorch.org/get-started/locally/
  # https://pytorch.org/get-started/previous-versions/
  # https://anaconda.org/pytorch/pytorch
  # https://anaconda.org/pytorch/torchvision
  # https://anaconda.org/pytorch/torchaudio
  # https://anaconda.org/pytorch/pytorch-cuda
  - pytorch::pytorch=2.0.1
  - pytorch::torchvision=0.15.2
  - pytorch::torchaudio=2.0.2
  - pytorch::pytorch-cuda=11.8
  # Check:
  # https://pytorch-geometric.readthedocs.io/en/latest/install/installation.html
  # https://anaconda.org/pyg/pyg
  - pyg::pyg=2.4.0
  - pip:
    - scikit_learn==1.3.0
    - hydra-nevergrad-sweeper==1.3.0.dev0
    - hydra-optuna-sweeper==1.2.0
    - sqlalchemy==1.4.44