- conda create --name grag python=3.9 -y
- conda activate grag
- conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.8 -c pytorch -c nvidia
- pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.0.1+cu118.html
- pip install peft
- pip install pandas
- pip install ogb
- pip install wandb
- pip install sentencepiece
- pip install torch_geometric
- pip install datasets
- pip install gensim==4.3.0
- pip install transformers==4.30.0
|
