【踩坑】PyG报错Undefined symbol: xxxxxx

2023-05-26 16:19:05 浏览数 (1)

首先确保pytorch、pyg、cuda的版本对应。

其次,所有步骤的安装,都是用pip。不要pip跟conda混用!!

比如,这里是PyTorch 2.0.1、CUDA11.7,

PyTorch:

代码语言:javascript复制
pip3 install torch torchvision torchaudio

PyG:

代码语言:javascript复制
pip3 install torch torchvision torchaudio
pip install torch_geometric
# Optional dependencies:
pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv -f https://data.pyg.org/whl/torch-2.0.0 cu117.html

有可能就可以了。

Installation — pytorch_geometric documentation Start Locally | PyTorch

0 人点赞