当前位置:   article > 正文

Linux下安装‘torch_geometric’、’torch-scatter’、 ‘torch-cluster’、 ‘torch-sparse’详细教程_linux服务器 conda install torch_geometric

linux服务器 conda install torch_geometric

要安装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==2.0.3
pip install -i https://pypi.tuna.tsinghua.edu
  • 1
声明:本文内容由网友自发贡献,转载请注明出处:【wpsshop博客】
推荐阅读
相关标签
  

闽ICP备14008679号