参考自https://blog.csdn.net/Andrew_zjc/article/details/117914736
下载pytorch
查阅自己电脑cuda版本
打开命令行,输入
nvidia-smi
可以看到我电脑cuda版本12.0,pytorch版本向下兼容
登陆torch官网下载,https://pytorch.org/
没有12.0,但由于向下兼容,这里选择11.8,复制命令行并开始下载
下载pytest-runner
pip install pytest-runner
下载pytorch-geometric
https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html
登陆官网下载
下载完成后即可