遇到的问题
pip install torch-sparse -f https://data.pyg.org/whl/torch-1.10.0+cu111.html
pip install torch-scatter -f https://data.pyg.org/whl/torch-1.10.0+cu111.html
上述命令执行之后出现timed out以及不适配,网络连接不稳定等问题。
问题分析
是因为从pip install中的whl文件无法获取成功,导致一直出错。
解决方案
1.查看自己需要的版本号(pytorch)
conda activate envname //进入conda环境(填写你自己的环境名称)
pip list //查看已经下载好的列表
记住自己的torch版本。
2.查看自己的CUDA版本
nvcc -V
3.进入网址(pytorch-geometric.com/whl/)寻找自己需要的版本号
找到对应的torch,里面的cu是你CUDA的版本,111表示CUDA11.1,点击进去。
4.找到对应的whl文件
这里面的cp表示的是你的python版本,Win和Linux自行选择即可。
5.安装包
从刚刚的conda环境中,使用cd命令进入你在浏览器中下载这些文件的文件夹位置下,执行下面的命令。
pip install torch_scatter-2.0.9-cp38-cp38-win_amd64.whl //例子
pip install torch_sparse-0.6.12-cp39-cp39-win_amd64.whl //例子
这样就可以快速安装这两个库了
这里还有一个小建议,最好是先装这两个包再安装
torch-geometric