【0基础可看】PyTorch-Geometric安装教程

自己记录以下

完整复现模型模型ner

conda info -e
conda activate name
conda install python==3.8.16

conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.3 -c pytorch

python
>>> import torch
>>> torch.__version__
>>>torch.cuda.is_available()

确定torch和cuda

已安装：查看torch和cuda版本

import torch
print(torch.__version__)  #注意是双下划线
print(torch.version.cuda)
print(torch.cuda.is_available())
print(torch.cuda.get_device_name())

未安装：安装我的版本号

# CUDA 11.3
conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.3 -c pytorch

安装好检查一下版本号

我的如下：
操作系统：Win10
Cuda版本：Cuda 11.3
Pytorch版本：Pytorch 1.12.0
Python版本：Python 3.8.13

pip install torch-scatter -f https://data.pyg.org/whl/torch-1.12.0+cu113.html
pip install torch-sparse -f https://data.pyg.org/whl/torch-1.12.0+cu113.html
pip install torch-cluster -f https://data.pyg.org/whl/torch-1.12.0+cu113.html
pip install torch-spline-conv -f https://data.pyg.org/whl/torch-1.12.0+cu113.html
pip install torch-geometric

如果这一步无法下载，超时等问题
直接点击这个链接，进行下载：https://pytorch-geometric.com/whl/torch-1.12.0+cu113.html
验证以下：

python -c "import torch_geometric"

出错

解决方法

pip install  torch-geometric -i http://pypi.douban.com/simple --trusted-host pypi.douban.com