【教程】安装torch_sparse、torch_cluster、torch_scatter、torch_spline

966 阅读1分钟

先进入这个链接,检查下pytorch版本对不对:

pytorch-geometric.com/whl/

pytorch官网:

Start Locally | PyTorch

由于我装的1.13.0太新了,所以降级装了1.12.1。

# conda create -n py39 python=3.9
# conda activate py39
pip3 install torch==1.21.1 torchvision==0.13.1 torchaudio==0.12.1

然后根据有没有GPU,进入对应的路径。

然后根据“python版本、系统类型”选择不同的whl并下载。

 然后执行安装:

wget https://data.pyg.org/whl/torch-1.12.0%2Bcu116/torch_sparse-0.6.15-cp39-cp39-linux_x86_64.whl
wget https://data.pyg.org/whl/torch-1.12.0%2Bcu116/torch_scatter-2.0.9-cp39-cp39-linux_x86_64.whl
wget https://data.pyg.org/whl/torch-1.12.0%2Bcu116/torch_cluster-1.6.0-cp39-cp39-linux_x86_64.whl
wget https://data.pyg.org/whl/torch-1.12.0%2Bcu116/torch_spline_conv-1.2.1-cp39-cp39-linux_x86_64.whl

pip install torch_sparse-0.6.15-cp39-cp39-linux_x86_64.whl
pip install torch_scatter-2.0.9-cp39-cp39-linux_x86_64.whl
pip install torch_cluster-1.6.0-cp39-cp39-linux_x86_64.whl
pip install torch_spline_conv-1.2.1-cp39-cp39-linux_x86_64.whl