安装PyG时踩的坑

今天在安装 PyG 的时候解决了以前困扰了一段时间的 bug。

OSError:找不到 libpython3.7m.so.1.0

解决方式:首先找到这个库存在的位置

find /home/<用户名> -name libpython3.7m.so.1.0

然后找到它存在的位置之后,在 ~/.bashrc 最后一行添加

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:<libpython3.7m.so.1.0的文件夹>

再执行

source ~/.bashrc

完美解决

GPU 不兼容

用 3090 的卡,没法用很老版本的 pytorch(但好多文章是用的老版本pytorch)

最终我找的这个版本(Python 3.7)

pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 \
	-f https://download.pytorch.org/whl/torch_stable.html

PyG 的 data 模块不兼容

老版本的 PyG 和新版本的模型似乎不兼容。

根据上文安装的 pytorch 版本,先安装:

pip install torch-scatter==2.0.7 torch-sparse==0.6.10 --no-index --no-cache-dir -f https://data.pyg.org/whl/torch-1.8.0%2Bcu111.html

再安装:

pip install torch-geometric==1.7.2 --no-cache-dir  

也可以选择性地安装:

pip install torch-cluster torch-spline-conv --no-index --no-cache-dir \
	-f https://data.pyg.org/whl/torch-1.8.0%2Bcu111.html

你可能感兴趣的:(python,pytorch)