1、首先确定机器使用的cuda版本nvcc -V

例如本服务器使用的时cuda11.1版本

2、创建一个conda环境:

conda create -n new_env python=3.8conda activate new_env

3、安装pytroch环境:
conda install pyg -c pyg
该命令会把一些常用的torch以及关联的工具包都安装到机器,验证是否正确安装:

如果发生错误:

需要自行去官网查找正确的版本:Previous PyTorch Versions | PyTorch

4、安装torch-geometric:
torch-geometric要和torch版本配套,登录网站链接: 链接查看对应的版本:


下载正确版本的组件,例执行pip install torch-sparse==0.6.12,其他的组件也一样

最后安装torch-geometricpip install torch-geometric
失败了可以尝试加上版本号pip install torch-geometric==2.0.4

5、验证torch-geometric:

成功导入即可