torch_geometric安装官网:
https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html

1. 查看CUDA版本:

方法一: 直接在终端检查版本
例如:

$ python -c "import torch; print(torch.version.cuda)"
>>> 11.1

方法二:通过命令:nvcc --version查看
例如:

$ nvcc --version

方法三:

>>>print(torch.version.cuda)
>>>print(torch.cuda.is_available())
结果:
11.1
True

如果没有安装cuda,参考以下网址安装:
https://blog.csdn.net/ashome123/article/details/105822040

2. 查看PyTorch版本:

$ python -c "import torch; print(torch.__version__)"
>>> 1.8.0
print(torch.__version__)

如果没有安装pytorch,则去官网安装,网址如下:
https://pytorch.org/get-started/locally/

注意:切记!切记!切记!

1.Pytorch和CUDA的版本一定要对应上,否则,即便安装了也会出错调用不了,一般情况下,不要安装最新版本Pytorch,因为CUDA和其他相关依赖库(安装torch_geometric,需要安装同时torch-scatter,torch-sparse,torch-cluster,torch-spline-conv等库)会跟不上,没有与之对应的版本,所以安装之前一定要核对好版本。
2.如果之前安装了其他版本的CUDA或者Pytorch,要卸载干净了再安装,如果没有卸载干净直接安装,系统就会有多个版本,就算用nvcc --version检测出来有安装该软件,也无法正常调用CUDA。

3.安装:

参考官网:https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html

pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-geometric

其中( T O R C H ) 替 换 为 P y T o r c h 的 版 本 号 , ( {TORCH})替换为PyTorch的版本号,( TORCH)替换为PyTorch的版本号,({CUDA})替换为上面查询得到的CUDA版本号。CUDA 版本号格式 (cpu, cu92, cu101, cu102, cu110, cu111)。例如对于PyTorch 1.8.0/1.8.1 and CUDA 11.1,安装命令如下:

pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.8.0+cu111.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.8.0+cu111.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-1.8.0+cu111.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.8.0+cu111.html
pip install torch-geometric

对于PyTorch 1.7.0/1.7.1 and CUDA 10.2,安装命令如下:

pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.7.0+cu102.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.7.0+cu102.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-1.7.0+cu102.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.7.0+cu102.html
pip install torch-geometric

4.检测是否安装成功:

$ python
Python 3.7.0 (default, Jun 28 2018, 13:15:42)
[GCC 7.2.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> from torch_geometric.data import DataLoader
>>>

若没报错,则安装成功。但我的没有直接安装成功,而是出现了下面的可能出现错误。

5.可能出现的错误:

按照以上安装步骤完成安装以后,还可能出现‘No module named 'torch_sparse’问题
那么,可以进入torch_sparse官网找对应的cuda、pytorch版本安装。
torch_sparse官网:https://github.com/rusty1s/pytorch_sparse
我的是PyTorch 1.8.0/1.8.1
To install the binaries for PyTorch 1.8.0 and 1.8.1, simply run

pip install torch-scatter torch-sparse -f https://data.pyg.org/whl/torch-1.8.0+${CUDA}.html

参考网址:

  1. 解决图神经网络库pytorch-geometric“版本不匹配”安装失败的问题
  2. 利用torch_geometric运行gcn
  3. No module named torch_sparse
  4. Linux下PyTorch、CUDA Toolkit 及显卡驱动版本对应关系(附详细安装步骤)
  5. Ubuntu 20.04 CUDA&cuDNN安装方法

如何安装torch_geometric?相关推荐

  1. 正确安装 torch_geometric库

    step1: 查看pytorch+cuda 版本 torch-scatter torch-sparse torch-cluster torch-spline-conv 这些关联包要与torch版本匹配 ...

  2. torch_geometric安装

    参考: 手把手教你安装torch_geometric库(pyg)_杰克船长有烦恼的博客-CSDN博客_torch安装 如何安装torch_geometric?_知行合一and至于至善的博客-CSDN博 ...

  3. PyTorch_Geometric 安装过程

    首先安装torch,如我们在这里安装1.6.0的GPU版本,在官网查找命令 在我这里的服务器中,使用conda直接运行命令,可自动配置所需的cuda版本 之后进行torch geometric的安装 ...

  4. 利用torch_geometric运行gcn

    pytorch出了图计算的工具torch_geometric后,gcn的实现就简单了,直接封装好了 首先需要安装torch_geometric $ pip install --no-index tor ...

  5. 图神经网络学习(一)-GCN及其应用

    内容提要:GCN背景简介+torch_geometric库安装+GCN处理Cora数据集 1.图神经网络 1.1 概念 原有的卷积神经网络主要用来解决欧式空间中的数据(数据规整,形状固定),例如图像数 ...

  6. 【无标题】Torch_geometric安装教程, 问题汇总详解。

    前情提要:本文只谈通过pip方式安装,conda安装的方式博主没有试过,感兴趣的小伙伴可以参考本文自己尝试一下. 首先,在核心上,Torch_geometric的安装方法几乎和pytorch一样,甚至 ...

  7. torch_geometric包安装

    torch_geometric包安装 torch安装 推荐whl安装,最不容易出错. 可先采用nvidia-smi命令检查cuda版本,安装对应版本的torch. [torch whl地址] pyg( ...

  8. 解决调用torch_geometric报错No module named ‘torch_sparse‘等问题,以及torch_sparse torch_scatter等的安装问题

    出现的问题:torch_geometric报错 会出现为torch_sparse torch_scatter等的问题 最近又开始搞图神经网络方面的东西,要用到torch_geometric.但是写好代 ...

  9. conda安装torch+torch_geometric

    @conda安装torch+torch_geometric 一. 基础命令 查看当前存在哪些虚拟环境 conda env list #查看当前存在哪些虚拟环境 或 conda info --envs ...

最新文章

  1. linux下定时清理磁盘日志步骤
  2. JAVA加密算法(DSA)
  3. 计算机与操作系统简介
  4. 初步认识Volatile-总结可见性的本质
  5. 苹果开发者账号那些事儿(三)
  6. JSP:Cookie实现永久登录(书本案例)
  7. 2016考研数学三复习计划
  8. Go 中的那些语法糖
  9. 微信邮箱登录入口,企业微信企业邮箱,企业邮箱登陆入口
  10. 蓝牙鼠标windows linux,解决Ubuntu 18.04与Windows 10双系统蓝牙鼠标连接的问题
  11. AutoCAD Civil 3D-加宽与超高
  12. Vue中添加背景图片
  13. HTML之在JavaScript中定义函数
  14. 基于SpringBoot的影视/短视频网站系统
  15. WPF中播放GIF图片
  16. android高德地图上加自定义菜单,自定义UI控件-UI界面定制-开发指南-Android 导航SDK | 高德地图API...
  17. 跨境电商如何找代运营公司?星之河
  18. 新型网络诈骗缘何层出不穷?
  19. DB-Day2 输入生日的月和日,计算出你下一个生日还有多少天
  20. Java、JSP基于JavaEE的家庭影院系统

热门文章

  1. 【博学谷学习记录】超强总结,用心分享|【Java Web】MySQL基础
  2. ext+dwr DynamicGridPanel 封装 态创建ext grid
  3. 用python写身高体重关系,excel表格输入数据进行查找
  4. 内网渗透(一)之基础知识-内网渗透介绍和概述
  5. bitblt简易使用教程
  6. 基于JAVA移动电商网站计算机毕业设计源码+系统+数据库+lw文档+部署
  7. 工商银行U盾使用笔记
  8. AbstractMethodError: javax.xml.parsers.DocumentBuilderFactory.setFeature(Ljava/lang/String;Z)V
  9. 使用 iCloud API 的正确姿势
  10. 读取txt文本数据,用区域生长方法分割求平面