Ubuntu系统安装torch_geometric

一、pip本地安装torch_scatter

官方地址:

https://pytorch-geometric.com/whl/

如上图,找到自己的torch和cuda版本 (例如torch-1.10.1+cu102就是指torch1.10.1版本,cuda10.2版本) 

然后找到对应的scatter版本(cp为python版本,Linux为Linux系统,选择x86_64为系统架构)

完成后下载到本地用命令安装(' '里面为路径):

pip install '/mnt/d/ZTT/Downloads/torch_scatter-2.1.0+pt112cu102-cp37-cp37m-linux_x86_64.whl'

安装完成后再安装torch_geometric

pip install torch-geometric

验证是否能正确导包,无报错即安装成功 

(places) tt@ZTT:~$ python
Python 3.7.16 (default, Jan 17 2023, 22:20:44)
[GCC 11.2.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch_geometric
>>> from torch_scatter import scatter_max
>>>

 

 

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值