ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

PyTorch Geometric(图卷积)环境安装 —— GPU

2021-11-01 23:01:28  阅读:284  来源: 互联网

标签:1.7 torch PyTorch cu110 pytorch https GPU Geometric geometric


1. 安装PyTorch

         进入 Pytorch 官网 pytorch.org,安装自己需要的版本。(我的是1.7.1版本  cuda11.0)

注意:Pytotch 最低为 1.4.0 以上

pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

 2. 安装PyTorch Geometric

        官方链接:Installation — pytorch_geometric 2.0.0 documentation

  • 安装相关依赖库  

pip install --no-index torch-scatter -f https://pytorch-geometric.com/whl/torch-1.7.1+cu110.html
pip install --no-index torch-sparse -f https://pytorch-geometric.com/whl/torch-1.7.1+cu110.html
pip install --no-index torch-cluster -f https://pytorch-geometric.com/whl/torch-1.7.1+cu110.html
pip install --no-index torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.7.1+cu110.html
  • 安装 torch-geometric

pip install torch-geometric -f https://data.pyg.org/whl/torch-1.7.1+cu110.html

注意:上述相关依赖库 也可以进入官网下载后离线安装。网址:https://pytorch-geometric.com/whl/

3. 测试

        没有报错就说明安装成功了

python -c "import torch_geometric"

公众号:清纯世纪 

标签:1.7,torch,PyTorch,cu110,pytorch,https,GPU,Geometric,geometric
来源: https://blog.csdn.net/qq_45100200/article/details/121090553

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有