安装PyTorch Geometric 一次成功

@安装PyTorch Geometric一次成功
首先确保你的torch已经安装
查看torch和cuda版本 我的是torch1.6.0 cuda 10.2
torch :命令行 python ->import torch -> torch.version
cuda : 命令行 python ->import torch -> torch.version.cuda

先下载这四个包在
依赖包
下载并安装下列包

pip install torch_spline_conv-latest+cu102-cp36-cp36m-linux_x86_64.whl
pip install torch_sparse-latest+cu102-cp36-cp36m-linux_x86_64.whl
pip install torch_scatter-latest+cu102-cp36-cp36m-linux_x86_64.whl
pip install torch_cluster-latest+cu102-cp36-cp36m-linux_x86_64.whl
#最后安装torch_geometric
pip install torch-geometric


本文来自互联网用户投稿,文章观点仅代表作者本人,不代表本站立场,不承担相关法律责任。如若转载,请注明出处。 如若内容造成侵权/违法违规/事实不符,请点击【内容举报】进行投诉反馈!

相关文章

立即
投稿

微信公众账号

微信扫一扫加关注

返回
顶部