PyTorch与PyTorch Geometric的安装过程
作者:子非闕 发布时间:2021-12-15 05:00:08
PyTorch与PyTorch Geometric的安装
GPU与CUDA,Python,PyTorch的匹配
1. 查看Linux系统中GPU的基础信息/NVIDIA Driver Version
nvidia-smi
nvidia-smi是nvidia 的系统管理界面 ,其中smi是System management interface的缩写,它可以收集各种级别的信息,查看显存使用情况。此外, 可以启用和禁用 GPU 配置选项 (如 ECC 内存功能)。
系统的Nvidia Driver Version决定着系统最高可以支持什么版本的cuda和cudatoolkit,Nvidia Driver是向下兼容的,详情如下(见Table 3. CUDA Toolkit and Corresponding Driver Versions):
2. 查看当前CUDA版本:
cat /usr/local/cuda/version.txt
我自己的环境最高可支持10.1版本的cuda和cudatoolkit,当前是10.0,版本向下兼容,并无什么问题。
3. CUDA Toolkit匹配PyTorch
CUDA Toolkit 和PyTorch的对应关系(见官网)
! 注意事项:服务器本身的CUDA版本与虚拟环境中安装的cudatoolkit包没有太大关系,一般安装pytorch时需要考虑的cuda版本指的应该是虚拟环境中安装的cudatoolkit包的版本
由于我需要用到v1.6.0的Pytorch,因此自己在虚拟环境里安装v10.1的CUDA Toolkit,系统GPU可接受最高版本v10.1。
torch与torchvision对应关系如下(详情见PyTorch / Vision):
因此,我需要安装的如下:
CUDA Toolkit == 10.1
Python == 3.7
PyTorch == 1.6
安装PyTorch
pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html
安装完成后可通过以下命令检查torch版本及对应的CUDA版本:
python -c "import torch; print(torch.__version__)"
python -c "import torch; print(torch.version.cuda)"
再通过以下命令查看GPU是否可用:
python
>>> import torch
>>> torch.cuda.is_available() # GPU是否可用
>>> torch.cuda.device_count()# GPU数量
>>> torch.cuda.current_device()# 当前GPU
>>> exit()
注意,GPU devices从0开始编号。
安装PyTorch Geometric
1. 快速安装
根据官网,如果PyTorch版本≥1.8.0,可以快速下载:
2. 自定义安装
自定义下载需要根据当前的PyTorch版本和CUDA版本下载相关的依赖,下载命令如下:
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-geometric
其中, ${TORCH}替换为当前环境下的PyTorch版本,目前支持1.4.0、1.5.0、1.6.0、1.7.0、1.7.1、1.8.0、1.8.1、和1.9.0; ${CUDA}替换为指定的CUDA版本,目前支持cpu、cu92、cu101、cu102、cu110和cu111。
例如对于PyTorch 1.6.0和CUDA 10.1:
pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.6.0+cu101.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.6.0+cu101.html
pip install torch-geometric
3. 版本依赖
使用自定义安装时,依然可能会出现安装失败的问题,因为pytorch_geometric几个相关库之间有比较强的依赖关系,建议是在自定义安装的基础上指定对应库的版本,例如对于pytorch1.6.0和cuda10.1:
pip install torch-scatter==2.0.5 -f https://pytorch-geometric.com/whl/torch-1.6.0+cu101.html
pip install torch-cluster==1.5.8 -f https://pytorch-geometric.com/whl/torch-1.6.0+cu101.html
pip install torch-sparse==0.6.7 -f https://pytorch-geometric.com/whl/torch-1.6.0+cu101.html
pip install torch-spline-conv==1.2.0 -f https://pytorch-geometric.com/whl/torch-1.6.0+cu101.html
pip install torch-geometric==1.6.1 -f https://pytorch-geometric.com/whl/torch-1.6.0+cu101.html
来源:https://blog.csdn.net/Fheonixw/article/details/123596839
猜你喜欢
- 优化糟糕设计的表结果或者索引能很大程度改进mysql的性能。 如果需要高性能, 那么就需要根据不同的操作需求精心设计表结构和索引, 这当然需
- 品牌是我们一直挂在嘴边的词语,视觉设计师们经常说到,公司的品牌该如何如何去设计?这个违背了我们的公司品牌!等等。之前我有谈过关于 品牌灵魂的
- 创建一个名为templatetags的python module。新建一个名为verbose_name.py的文件。from django
- 在使用Django做前端后端项目时,登陆认证方法往往使用的是jwt_token,但是想自定义登陆成功和失败的返回体。1.当用户名和密码正确就
- 教你配置属于自己的PYcharm界面色彩风格,PYthon学习必备 GO第一步,换成深色背景,保护视力PyCharm默认的背景是白色的,比较
- 前言在Django应用程序中发送电子邮件最常见的用例是密码重置、帐户激活和发送与您的应用程序相关的一般通知。下面来看看详细的介绍吧。配置Dj
- 从业这几年,自己写过的和帮人参谋的所谓“设计规范”不少了,这个东西大概在中国的决策层眼里是这么回事儿 - 一帮农民在一块田里种粮食,起先天气
- 前言python2.x版本的字符编码有时让人很头疼,遇到问题,网上方法可以解决错误,但对原理还是一知半解,本文主要介绍 python 中字符
- 在TypeScript 项目中,我们经常需要使用声明一系列的ts类型。然而,手动写的效率实在太低,编写一个自动生成ts类型的工具可以解放生产
- 目录项目地址:前言使用方式主要代码blive_record.pyconfig.py(配置文件)项目地址:https://github.com
- 如何在线更改密码?<%id = Request("id")newpassword =
- ndarray.ndim:维度ndarray.shape:形状ndarray.size:元素个数ndarray.dtype:元素数据类型nd
- 一:从版本3.23升级到4.104.10新特性:支持事务处理和存储过程升级过程的注意事项:1:升级采用直接复制安装文件的方法,简单实用,但要
- 目录查找算法 -- 简介顺序查找二分查找插值查找斐波那契查找树表查找1、二叉树查找算法。2、平衡查找树之2-3查找树(2-3 Tree)3、
- 本文较为详细的介绍了python中raw_input的用法,使用raw_input 能够很方便的丛控制台读入数据。具体用法示例如下:1.输入
- 前言几乎每个程序都需要用到图片。下面就来给大家介绍前端+PHP后端实现微信小程序实现图片上传功能,分享出来供大家参考学习,下面话不多说了,来
- 最近用layer ui上传文件遇到了一个问题,我想在上传文件之前把data-id传入后台,layer文档找了一下也没有找到类似的说明,经过一
- Real Numbers实数实数是具有小数部分的数字, 当然, 实数不是专门用来表示小数的, 也可以用DECIMAL来存储那些无法用INTE
- JavaScript组件打包模式js组件通常带着css image ,但这样使用起来可能会有些小麻烦,为了让组件足够的solo,有了把css
- 一、绘图命令操纵海龟绘图有很多命令,可以划分为三种:画笔运动命令、画笔控制命令、全局控制命令1、画笔运动命令命令说明turtle.forwa