ToB企服应用市场:ToB评测及商务社交产业平台

标题: 安装torch-geometric库 [打印本页]

作者: 我可以不吃啊 时间: 2024-12-30 14:02
标题: 安装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版本

nvcc -V

复制代码

查看pytorch版本

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 下载链接

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 安装

pip install wheel文件地址

复制代码

3.下载并安装torch-geometric库

网址：PyG

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

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

复制代码

安装完成，无报错

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

欢迎光临 ToB企服应用市场:ToB评测及商务社交产业平台 (https://dis.qidao123.com/)