5行代码完成torch_geometric安装

首先查看自己安装的torch版本和cuda对应的版本,根据自己的版本号替换每个命令的粗体部分,其他部分不变,依次执行下面1-5条命令(我的torch版本为1.12.1,cuda版本是11.3,如下图所示)

5行代码完成torch_geometric安装_第1张图片

1、pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.12.1+cu113.html

2、pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.12.1+cu113.html

3、pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-1.12.1+cu113.html
4、pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.12.1+cu113.html
5、pip install torch-geometric

你可能感兴趣的:(python学习,pytorch,python)