最近需要使用到PyG框架,安装的时候需要注意一些问题,记录一下,方便后来者避坑!
步骤1
首先要先确定自己的torch版本
如果使用的Anaconda可以使用conda list
命令查看版本号
进入官网https://pytorch-geometric.com/whl/找到与torch对应的torch-geometric,cu表示cuda版本,cpu表示cpu版本,选择对应的进去。
cuda版本号可以在控制台使用nvcc --version
来查看
![在这里插入图片描述](https://img-blog.csdnimg.cn/85b4808f91a2454d9c2db4ed8b5e2a8e.png
步骤2
之后开始安装依赖,主要包括torch-cluster, torch-scatter, torch-sparse
cp表示的是python版本,根据自己需要选择linux或win版本
下载完成whl文件之后&