欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  网络运营

Linux下安装‘torch_geometric’、’torch-scatter’、 ‘torch-cluster’、 ‘torch-sparse’详细教程

程序员文章站 2022-03-11 10:12:28
要安装torch_geometric,必须按照以下顺序进行!!!第一步,必须安装torch,本人使用cuda9.2,使用命令安装的1.4.0版本,即conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=9.2 -c pytorch,可以再Pytorch官网选择自己需要的版本安装命令进行安装。第二步,再安装’torch-scatter’、 ‘torch-cluster’、 'torch-sparse’模块。即:pip install...

要安装torch_geometric,必须按照以下顺序进行!!!

方法一:直接安装

这种方法安装的时候可能比较慢!!如果等不及的可以参照方法二安装。

  • 第一步,必须安装torch,本人使用cuda9.2,使用命令安装的1.4.0版本,即conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=9.2 -c pytorch,可以再Pytorch官网选择自己需要的版本安装命令进行安装。
  • 第二步,再安装’torch-scatter’、 ‘torch-cluster’、 'torch-sparse’模块。即:
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch-scatter

本文地址:https://blog.csdn.net/weixin_43994864/article/details/110468033