安装torch-geometric库

目次
1.查看 torch 和 CUDA 版本
2.依次下载和 torch 和 CUDA 对应版本的四个依赖库pyg-lib、torch-scatter、torch-sparse、torch-cluster以及torch-spline-conv
3.下载并安装torch-geometric库

---

1.查看 torch 和 CUDA 版本

查看CUDA版本

1. nvcc -V

复制代码

查看pytorch版本

1. pip show torch

复制代码

2.依次下载和 torch 和 CUDA 对应版本的四个依赖库pyg-lib、torch-scatter、torch-sparse、torch-cluster以及torch-spline-conv

 网址： https://data.pyg.org/whl/
 根据自己的torch版本和CUDA选择

选择必要的版本，右键复制链接。

在下令行使用 wget 下载链接

1. wget -P 目标地址 https://data.pyg.org/whl/torch-2.0.0%2Bcu118/torch_sparse-0.6.18%2Bpt20cu118-cp310-cp310-linux_x86_64.whl

复制代码

下载以上五个库的wheel文件之后，使用 pip install 安装

1. pip install wheel文件地址

复制代码

3.下载并安装torch-geometric库

网址：PyG

 

根据自己的torch和cuda版本安装 torch-geometric库

1. pip install torch_geometric -f https://data.pyg.org/whl/torch-{torch版本号}+cu{CUDA版本号}.html
2. # pip install torch_geometric -f https://data.pyg.org/whl/torch-2.0.0+cu118.html

复制代码

安装完成，无报错

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。