栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

安装pytorch-geometric

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

安装pytorch-geometric

安装pytorch-geometric

pip install pytorch-geometric
pip install pytorch-geometric-temporal

完了之后,运行代码,提示

FileNotFoundError: Could not find module 'C:SoftwaresAnacondaenvspytorch_spspmm_cuda.pyd

原因是,安装的torch-sparse 、torch-spline-conv、torch-scatter、torch-cluster版本不对。

查看好自己的cuda版本,建议用这个查:

import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.backends.cudnn.version())

之后访问:https://pytorch-geometric.com/whl/
看仔细自己的torch+cuda版本,点击跳转,下载对应python版本的whl文件,离线安装。
这四个库不建议在线,安装。

到这应该就解决了。

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/840238.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号