PyTorch_Geometric 安装过程

首先安装torch,如我们在这里安装1.6.0的GPU版本,在官网查找命令
PyTorch_Geometric 安装过程_第1张图片
在我这里的服务器中,使用conda直接运行命令,可自动配置所需的cuda版本

之后进行torch geometric的安装
其有四个前置包,找到官网地址 https://pytorch-geometric.com/whl/torch-1.6.0.html
这里的1.6.0可改为自己torch的对应版本
PyTorch_Geometric 安装过程_第2张图片
选择对应的包(cluster、scatter、spline、sparse)下载到本地或者服务器,我所下载的如下在这里插入图片描述
进入下载文件夹路径,然后使用pip install 进行安装,如

pip install torch_cluster-latest+cu102-cp38-cp38-linux_x86_64.whl

安装完四个后,再安装torch_geometric

pip install torch_geometric

完成
在这里插入图片描述

你可能感兴趣的:(工具,Python杂七杂八,深度学习,pytorch,深度学习,python)