from torch._c import *找不到指定的模块解决

from torch._c import *找不到指定的模块解决方法

可能的原因,你的torch安装的不对,不能 pip install torch
版本问题
CUDA11.0的服务器安装torch_geometric
在终端输入nvidia-smi,看右上角的CUDA Version对应的版本

显示如下:

二、新建环境进行相关包的安装
1.新建环境:conda create -n xxx python=3.8 (注:xxx为你自己新建环境名字,可以自己取,后面都用xxx代替)

2.激活环境:因为服务器一般是Linux,所以使用命令 source activate xxx 成功后自己用户名前会出现(xxx)

3.安装pytorch,建议安装新版的,pytorch链接,我这里安装的是pytorch1.10.0,但是要注意,改掉自己对应的cuda版本,如pytorch1.10.0安装指令为

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
我的CUDA是11.0,所以指令改为:

conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch
等待安装完成

4.安装geometric:链接在这:torch_geometric链接。因为这里我们的pytorch版本大于1.8,安装官网指示

我们直接运行:conda install pyg -c pyg conda-forge 就可以了,等待安装完成就可以使用了

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值