安装PyTorch Geometric 一次成功

it2023-01-12  76

@安装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
最新回复(0)