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