Py-TorchGeometric工具安装与配置(含PyTorch安装)
Py-TorchGeometric工具安装与配置(含PyTorch安装)

Py-TorchGeometric工具安装与配置(含PyTorch安装)

官网信息

官方文档:https://pytorch-geometric.readthedocs.io/en/latest/

不要pip安装,打开github:https://github.com/pyg-team/pytorch_geometric

找到:

image-20240126135422006

image-20240126144405356

安装torch

根据cuda版本安装torch:

image-20240126140237265

https://pytorch.org/get-started/previous-versions/

image-20240126150846895

pip install torch==1.12.1+cu102 torchvision==0.13.1+cu102 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu102

torch-geometric安装

torch-geometric要和torch版本配套

image-20240126152015238

每个包都装;尽量安装新版;cp指python的版本;

image-20240126152741094

下载后,用pip安装:

image-20240126153116951

最后安装torch-geometricpip install torch-geometric

发表回复

您的电子邮箱地址不会被公开。