kickoff-ml-project/environment.yml at main · quy-ng/kickoff-ml-project · 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
name: ml-venv
channel_priority: disabled
channels:
  - pytorch
  - nvidia
  - pyg
  - conda-forge
dependencies:
  - python=3.12
  - pip
  - blas
  - networkx
  - numpy
  - scipy
  - pandas
  - scikit-learn
  - matplotlib
  - jupyterlab
  - ipywidgets
  - panel
  - pytest
  - pytorch::pytorch=2.4.1
  - pytorch::pytorch-cuda=12.4
  - pytorch::torchvision=0.19.1
  - pytorch::torchaudio=2.4.1
  - pyg=2.6.1
  - transformers=4.46.0
  - sentencepiece
  - lightning
  - click
  - wandb
  - dvc
  - torchmetrics
  - evaluate
  - pip:
    - pyg_lib==0.4.0+pt24cu124
    - torch_scatter==2.1.2+pt24cu124
    - torch_sparse==0.6.18+pt24cu124
    - torch_cluster==1.6.3+pt24cu124
    - torch_spline_conv==1.2.2+pt24cu124
    - -f https://data.pyg.org/whl/torch-2.4.0+cu124.html