PyG文档之一:安装
PyG安装
- 1.GPU安装
- 1.1查看torch,CUDA版本
- 1.2根据版本号安装支持库
- 2.CPU安装(推荐新手)
- 2.1 新建虚拟环境(若是初次安装,此步骤也可跳过)
- 2.2 安装cpu版torch并查看版本
- 2.3 根据版本安装支持库
- 3.检查安装结果
1.GPU安装
pyg是torch-geometric的简称,类似于numpy,pyg已经成为了图神经网络的一大杀器。安装pyg需要预先安装torch,如需使用GPU,则还需要安装cuda。这两者的安装很简单,不再赘述。初学者建议只安装cpu版本,会简单一些。最好再安装一个Anaconda来管理虚拟环境。
我的演示环境:win10+conda+notebook,显卡是MX150
Linux下则默认你已经会安装了。
相关URL:
cuda官方网站
torch官方网站
pyg官方文档
1.1查看torch,CUDA版本
#$表示窗口命令
$ python -c "import torch; print(torch.__version__)"
>>> 1.6.0
$ python -c "import torch; print(torch.version.cuda)"
>>> 10.2
1.2根据版本号安装支持库
pip install torch-scatter==latest+cu102 -f https://pytorch-geometric.com/whl/torch-1.6.0.html
pip install torch-sparse==latest+cu102 -f https://pytorch-geometric.com/whl/torch-1.6.0.html
pip install torch-cluster==latest+cu102 -f https://pytorch-geometric.com/whl/torch-1.6.0.html
pip install torch-spline-conv==latest+cu102 -f https://pytorch-geometric.com/whl/torch-1.6.0.html
pip install torch-geometric
如果你的版本是cuda92+torch1.5.0,则你应该使用如下安装语句:
$ pip install torch-scatter==latest+cu92 -f https://pytorch-geometric.com/whl/torch-1.5.0.html
其它的以此推类。如果遇到找不到对应包的错误,则建议换一个torch或者cuda的版本。
我一开始是cuda92,但是安装后出现了问题。推荐版本:cuda:10.2 torch:1.5/1.6
2.CPU安装(推荐新手)
若使用CPU安装,则需要安装cpu版本的pytorch,为了不破坏原有环境,建议新建一个虚拟环境。
2.1 新建虚拟环境(若是初次安装,此步骤也可跳过)
$ conda create -n pyg_cpu
#激活虚拟环境
$ conda activate pyg_cpu
2.2 安装cpu版torch并查看版本
$ conda install pytorch torchvision cpuonly -c pytorch
$ python -c "import torch; print(torch.__version__)"
>>> 1.6.0
2.3 根据版本安装支持库
pip install torch-scatter==latest+cpu -f https://pytorch-geometric.com/whl/torch-1.6.0.html
pip install torch-sparse==latest+cpu -f https://pytorch-geometric.com/whl/torch-1.6.0.html
pip install torch-cluster==latest+cpu -f https://pytorch-geometric.com/whl/torch-1.6.0.html
pip install torch-spline-conv==latest+cpu -f https://pytorch-geometric.com/whl/torch-1.6.0.html
pip install torch-geometric
3.检查安装结果
import torch
import torch_geometric
若无报错,则说明安装成功,若有报错,首先检查一下安装步骤,是否做到版本一致。
另外cuda的版本有可能显示的不一定是真实安装的版本。建议将cuda全部删除,然后再安装一个新的。
PyG文档之一:安装相关推荐
- oracle12c配置文档,Oracle12C安装配置文档
Oracle12C安装配置文档 Oracle12C安装配置文档 准备软件: 开始安装: 打开从官网下载下来的两个压缩包,进行解压 打开解压好的后缀为2of2的文件夹 找到路径为database下的&q ...
- Unity离线文档的安装与加速
Unity离线文档的安装与加速 总览 平时在使用Unity时, 经常会查询一些Api或者用法示例, 但是Unity官网经常出问题, 而且有时访问速度不够理想, 我们可以将文件下载下来在本地查看, Un ...
- NumPy 文档与安装 NumPy
NumPy 文档与安装 NumPy 1. NumPy https://numpy.org/ NumPy brings the computational power of languages like ...
- PyG文档之二:快速入门
快速入门 图数据常规操作 栗子1:实例化一个data 栗子2:Data的一些内置函数 标准数据集 栗子3:载入标准数据集 小批量训练 栗子4:PyG的DataLoader 数据转换 栗子5:PyG 的 ...
- 如何在linux部署pdf文档,LINUX安装部署文档.pdf
LINUX 安装部署详解文档 文档内容简介: 一:详解安装前的准备工作---------------------- 二:安装LINUX ---------------------------... 三 ...
- Office Online Server 在线编辑Office文档,安装部署
一.Office Online Server介绍 Office Online Server是 Office Web Apps Server 的升级版本,安装环境必须为两台Windows Server ...
- 基于pdf2docx模块Python实现批量将PDF转Word文档(安装+完整代码教程)
PDF文件是一种常见的文档格式,但是在编辑和修改时不太方便,因为PDF本质上是一种静态的文档格式.因此,有时候我们需要将PDF文件转换成Word格式,以便更好地编辑和修改文档.在本篇文章中,我们将介绍 ...
- 清软英泰plm服务器安装文档,操作文档集合 (安装和管理 AnswerBook2 服务器 - zh)
操作文档集合 AnswerBook2 集合 是一个逻辑书组.例如,您可有一个包括所有 SolarisTM 面向开发者的书的集合.此集合可能是一个 AnswerBook2 (SGML) 集合或一个 An ...
- mysql安装文档_mysql安装文档
linux下使用yum安装mysql --安装的mysql版本为mysql55-mysql-server.x86_64 1.安装 查看有没有安装过: yum list installed mysql* ...
最新文章
- 爬虫系列之----Requests库
- CentOS下TPC-W安装
- python3.7安装pip问题_python3.7安装, 解决pip is configured with locations that require TLS/SSL问题...
- php phpunit selenium,phpunit+selenium环境筹建
- win10 子系统 ubuntu 16.04 安装 docker
- MFC学习之路之多媒体 --(1) DirectShow
- 明天一定要记得放一个硬币在口袋
- 【性能测试】Linux系统监控-Top命令
- 【华为云技术分享】详解浏览器跨域的几种方法
- Maven私服(二) - Nexus的安装
- 人工智能还能怎么玩?谷歌反手就是一个红狮子雕塑
- 股票大作手操盘术---到手的利润
- macOS Ventura 13.0 (22A380) 正式版发布,ISO、IPSW、PKG 下载
- BLE安全之SM剖析(3)
- [论文阅读] (23)恶意代码作者溯源(去匿名化)经典论文阅读:二进制和源代码对比
- 【DevOps】总结下容器方式构建Grafana-reporter生成PDF格式报告
- PYTHON将成绩从百分制变换到等级制
- java ajax 联动菜单_二级联动菜单代码(AJAX JAVA).
- AdSense 介绍
- 图的邻接矩阵表示法及顶点入度、出度的计算方法