torch-scatter、torch-sparse、torch-cluster、torch-spline-conv等包,直接用pip安装会失败
比如使用 pip install torch_scatter
安装torch_scatter会失败。
网上有的解决方法说是,在后面加上 -f https://pytorch-geometric.com/whl/torch-1.x.x.html ,但是试了还是失败
最后解决方法是直接用whl文件安装
1、访问https://pytorch-geometric.com/whl/
选择与本地环境对应的torch版本,比如我的是1.7.1。
https://pytorch-geometric.com/whl/torch-1.7.1.html
2、找到和本地环境对应的scatter/sparse/cluster/spline-conv版本
文件名主要分为五段式:
1:包名;2:版本;3:cuda版本,如cuda10.1可以选择cu101;4:python版本,python3.6可以选择cp36;5:操作系统信息。这里3、4、5需要和自己的环境对应!
3、找到对应版本的whl文件后,可以下载下来本地安装,也可以右键复制文件下载链接后直接安装
安装命令:pip install “whl文件的地址”
如:
pip install https://data.pyg.org/whl/torch-1.7.0/torch_scatter-2.0.5%2Bcpu-cp36-cp36m-win_amd64.whl
即可安装成功