ULTRA / setup /Setup.txt
niobures's picture
ULTRA (code, datasets, models, papers, setup)
a9c9942 verified
Raw History Blame Contribute Delete
861 Bytes
Ultra PyG реализован с помощью Python v3.9, PyTorch v2.1 и PyG v2.4 (CUDA v11.8 или более поздней версии при работе на GPU).
> Python v3.9.11 (https://www.python.org/ftp/python/3.9.11/python-3.9.11-amd64.exe)
Создать системную переменную окружения CUDA_HOME=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.3
(возможно, эта переменная также называется CUDA_PATH)
pip install torch==2.1.0 --index-url https://download.pytorch.org/whl/cu118
(попробовать установить pip install torch==2.2.0 --index-url https://download.pytorch.org/whl/cu121)
pip install torch-scatter==2.1.2 torch-sparse==0.6.18 torch-geometric==2.4.0 -f https://data.pyg.org/whl/torch-2.1.0+cu118.html
pip install ninja easydict pyyaml