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 就可以了,等待安装完成就可以使用了