- 首先自主安装cuda10.1和对应的cudnn
- 接下来参考 https://www.jianshu.com/p/8dcb4af749fd
对链接文章的补充:
(1)第7步安装4个whl文件:
torch_cluster-1.5.2+cu101-cp36-cp36m-win_amd64.whl
torch_scatter-2.0.4+cu101-cp36-cp36m-win_amd64.whl
torch_sparse-0.6.0+cu101-cp36-cp36m-win_amd64.whl
torch_spline_conv-1.2.0+cu101-cp36-cp36m-win_amd64.whl
(可以在浏览器搜索包名直接下载)
(也可以下网盘链接:https://pan.baidu.com/s/1eOmlsokpH4L8Ij35s4uJmA 提取码:vfs9 )
(2)最后加上 pip install torch-geometric 才能完成整个下载哦
pip install torch-geometric