前言:
前两天原想在笔记本上安装GPU版的pytorch来使用torch_geometric,删了原来的torch,再安装就出现各种问题。找不到torch_scatter.scatter_cpu、不能调用安装后又torch_scatter的函数、C++工具等等各种各样的问题。后面发现是torch、torch_scatter、torch_sparse的版本冲突。最后使用torch-1.4(GPU),torch_scatter(2.0),torch_sparse(0.6)、torch_geometric(1.4)的whl文件解决了问题。
总结:
查了很多网上教程,适合自己的比较少,一定要多看文档。
少随便更改包版本!!!
whl大法真的好!!!
安装:
【setup.py 文件命令】python setup.py install
【whl文件命令】pip install 包名.whl
1、清华镜像
https://pypi.tuna.tsinghua.edu.cn/simple/
2、Unofficial Windows Binaries for Python Extension Packages
【whl】https://www.lfd.uci.edu/~gohlke/pythonlibs/#cvxopt
3、pytorch
【官网】https://pytorch.org
【离线安装包whl】https://download.pytorch.org/whl/torch_stable.html
4、torch-sparse离线安装包
【tar.gz】https://pypi.tuna.tsinghua.edu.cn/simple/torch-sparse/
【whl】https://pytorch-geometric.com/whl/torch-1.4.0.html(外网,下同)
5、torch-scatter离线安装包
【tar.gz】https://pypi.tuna.tsinghua.edu.cn/simple/torch-scatter/
【whl】https://pytorch-geometric.com/whl/torch-1.4.0.html
【github】https://github.com/rusty1s/pytorch_scatter
6、torch-spline-conv离线安装包
【whl】https://pytorch-geometric.com/whl/torch-1.4.0.html
6、torch-geometrics
【官方文档】https://pytorch-geometric.readthedocs.io/en/latest/index.html
【离线安装包tar.gz】https://pypi.tuna.tsinghua.edu.cn/simple/torch-geometric/
【开源模型】https://github.com/rusty1s/pytorch_geometric
【github】https://github.com/rusty1s/pytorch_geometric
版权声明:本文为CSDN博主「Hi Daniel」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/u013313168/article/details/104793177