0
点赞
收藏
分享

微信扫一扫

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

归零者245号 2022-05-01 阅读 171

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 条评论