PyG吐血安装

最新推荐文章于 2024-08-22 14:16:01 发布

JJun_

最新推荐文章于 2024-08-22 14:16:01 发布

阅读量866

点赞数 1

文章标签： pip pytorch

本文链接：https://blog.csdn.net/qq_40984119/article/details/114949418

版权

PyG吐血安装

前言

前言

学习GCN会用到torch_geometric这个库，于是进行下载。其实一件挺简单的事，硬是搞了半天。以下是两种安装方法。
环境：python 3.6 + torch 1.6.0

一、官方文档

地址：https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html.

下载依赖库

pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-geometric

where ${CUDA} and ${TORCH} should be replaced by your specific CUDA version (cpu, cu92, cu101, cu102, cu110, cu111) and Py