Linux本地安装包安装PyG

基础环境

Linux系统
python 3.6.13
torch 1.5.0
cuda 10.1
cudnn 7.6.3

确保安装的torch和cuda版本配套，可在torch下载官网查看，注意系统不同命令也不同
检查cuda版本命令：

nvcc -version
>>nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Fri_Feb__8_19:08:17_PST_2019
Cuda compilation tools, release 10.1, V10.1.105

python -c "import torch; print(torch.version.cuda)"
>>10.1

安装包下载

pytorch-geometric官网地址 对应PyTorch1.5，下载时根据torch版本选择，如我的python3.6，torch1.5.0，cuda10.1，linux系统对应的whl

其它版本的torch只需要将链接中的1.5换成对应版本，如1.4
https://pytorch-geometric.com/whl/torch-1.4.0.html

安装

进入存放whl文件的文件夹，pip安装命令：

pip install torch_scatter-latest+cu101-cp36-cp36m-linux_x86_64.whl
pip install torch_sparse-latest+cu101-cp36-cp36m-linux_x86_64.whl
pip install torch_cluster-latest+cu101-cp36-cp36m-linux_x86_64.whl
pip install torch_spline_conv-latest+cu101-cp36-cp36m-linux_x86_64.whl
pip install pytorch_geometric

检查安装结果

>>import torch
>>import torch_geometric

至此完成，之所以选择本地安装是因为直接使用命令安装出现错误，具体原因没有了解，之后有机会补上。

ERROR: Exception: Traceback (most recent call last): File
“/home/ssr/miniconda3/envs/py36tc/lib/python3.6/site-packages/pip/_vendor/resolvelib/resolvers.py”,
line 171, in _merge_into_criterion
crit = self.state.criteria[name] KeyError: ‘torch-scatter’

During handling of the above exception, another exception occurred:

Traceback (most recent call last): File
“/home/ssr/miniconda3/envs/py36tc/lib/python3.6/site-packages/pip/_internal/cli/base_command.py”,
line 189, in _main
status = self.run(options, args) ’