安装torch_sparse、torch_cluster、torch_scatter、torch_spline

2022-11-14 16:58:34 浏览数 (1)

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

https://pytorch-geometric.com/whl/

pytorch官网:

Start Locally | PyTorch

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

代码语言:javascript复制
# 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并下载。

 然后执行安装:

代码语言:javascript复制
wget https://data.pyg.org/whl/torch-1.12.0+cu116/torch_sparse-0.6.15-cp39-cp39-linux_x86_64.whl
wget https://data.pyg.org/whl/torch-1.12.0+cu116/torch_scatter-2.0.9-cp39-cp39-linux_x86_64.whl
wget https://data.pyg.org/whl/torch-1.12.0+cu116/torch_cluster-1.6.0-cp39-cp39-linux_x86_64.whl
wget https://data.pyg.org/whl/torch-1.12.0+cu116/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

0 人点赞