Pytorch/PyG 踩坑整理

一、PyG 安装指北

1、pip安装1.4.0版本的pytorch，去PyPi下载wheel后本地安装（pypi版本无需指定cuda版本）

2、测试 torch.cuda.is_available() 是否为True

3、去 pytorch-geometric.com/whl/torch-1.4.0.html 下载cu101+linux版本的torch_sparse/torch_cluster/torch_scatter/torch_spline_conv，pip安装

4、测试 import torch_sparse，如果出现libcusparse.so.10不存在问题，则安装cudatoolkit：conda install cudatoolkit=10.1；如果出现GLIBCXX不存在问题，首先查看libstdc++.so.6中是否有该GLIBCXX：strings conda-env-path>/lib/libstdc++.so.6 | grep GLIBCXX，如果存在则添加到环境变量 export LD_LIBRARY_PATH=<conda-env-path>/lib:$LD_LIBRARY_PATH，如果不存在则需要安装GCC 6与libgcc，具体见github: torch_geometric: issue 1001

5、pip install torch_geometric

6、安装完毕

二、环境配置相关问题

Bug: [W NNPACK.cpp:80] Could not initialize NNPACK! Reason: Unsupported hardware.

问题描述：在使用服务器的登录节点进行inference时发生的错误。

解决方案：源代码重装Pytorch，使用环境变量 USE_NNPACK=0。

来源：Bug: [W NNPACK.cpp:80] Could not initialize NNPACK! Reason: Unsupported hardware - #2 by ptrblck - PyTorch Forums