今天在安装 PyG 的时候解决了以前困扰了一段时间的 bug。
解决方式:首先找到这个库存在的位置
find /home/<用户名> -name libpython3.7m.so.1.0
然后找到它存在的位置之后,在 ~/.bashrc 最后一行添加
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:<libpython3.7m.so.1.0的文件夹>
再执行
source ~/.bashrc
完美解决
用 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 和新版本的模型似乎不兼容。
根据上文安装的 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