源码安装pytorch_scatter, pytorch_cluster

动机

笔者使用清华镜像站作为pip源,直接pip install torch_scatterpip install torch_cluster安装不成功,考虑使用源码安装。下面以安装pytorch_cluster-1.4.5为例说明源码安装过程,pytorch_scatter的源码安装方式类似。
注意,安装pytorch_scatter, pytorch_cluster需要预先安装对应版本的pytorch

下载源码

可以从github找到对应版本的源码下载,也可以观察pip install torch-cluster==1.4.5的输出

Collecting torch-cluster==1.4.5 (from -r requirements.txt (line 75))
  Using cached https://pypi.tuna.tsinghua.edu.cn/packages/c3/70/1d827d6fd1e03bb5ae84852dd0070c6574105c37e7b935284f6e990932db/torch_cluster-1.4.5.tar.gz

Ctrl-C结束pip install ttorch-cluster==1.4.5进程,手动下载源码压缩包

wget https://pypi.tuna.tsinghua.edu.cn/packages/c3/70/1d827d6fd1e03bb5ae84852dd0070c6574105c37e7b935284f6e990932db/torch_cluster-1.4.5.tar.gz

源码安装

解压下载下来的源码压缩包

tar zxvf torch_cluster-1.4.5.tar.gz

进入解压出来的torch_cluster-1.4.5目录下安装,注意安装过程需要联网,因为setup.py会从网络上下载

cd torch_cluster-1.4.5
python setup.py install

如果pytorch, gcc的版本都适配,应该能够安装成功

测试

离开源码安装目录torch_cluster-1.4.5,执行代码

python -c "import torch_cluster"

如无报错,则安装成功。注意一定要离开源码安装目录,因为在源码目录下,无需安装,也是存在torch_cluster的模块的,此时执行python -c "import torch_cluster"成功不代表pytorch_cluster包安装成功

你可能感兴趣的:(Python,pytorch_scatter,pytorch_cluster,安装)