手把手教你做网站7网站制作建设公司

目录

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

安装完成，无报错