如何判断pytorch是否支持GPU加速


Posted in Python onJune 01, 2021

加上这句代码:

print torch.cuda.is_available()

如何判断pytorch是否支持GPU加速

判断完毕!说说在pytorch中如何查看gpu信息吧~

为什么将数据转移至GPU的方法叫做.cuda而不是.gpu,就像将数据转移至CPU调用的方法是.cpu?这是因为GPU的编程接口采用CUDA,而目前并不是所有的GPU都支持CUDA,只有部分Nvidia的GPU才支持。

PyTorch未来可能会支持AMD的GPU,而AMD GPU的编程接口采用OpenCL,因此PyTorch还预留着.cl方法,用于以后支持AMD等的GPU。

torch.cuda.is_available()

cuda是否可用;

torch.cuda.device_count()

返回gpu数量;

torch.cuda.get_device_name(0)

返回gpu名字,设备索引默认从0开始;

torch.cuda.current_device()

返回当前设备索引;

如何判断pytorch是否支持GPU加速

以上为个人经验,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
tornado框架blog模块分析与使用
Nov 21 Python
Python发送form-data请求及拼接form-data内容的方法
Mar 05 Python
Python及PyCharm下载与安装教程
Nov 18 Python
TensorFlow平台下Python实现神经网络
Mar 10 Python
jupyter notebook引用from pyecharts.charts import Bar运行报错
Apr 23 Python
python程序输出无内容的解决方式
Apr 09 Python
python 使用csv模块读写csv格式文件的示例
Dec 02 Python
python不同版本的_new_不同点总结
Dec 09 Python
史上最详细的Python打包成exe文件教程
Jan 17 Python
Matplotlib可视化之添加让统计图变得简单易懂的注释
Jun 11 Python
python代码实现备忘录案例讲解
Jul 26 Python
Pygame Time时间控制的具体使用详解
Nov 17 Python
pytorch 两个GPU同时训练的解决方案
Jun 01 #Python
使用Django实现商城验证码模块的方法
Jun 01 #Python
pytorch通过训练结果的复现设置随机种子
Jun 01 #Python
matplotlib画混淆矩阵与正确率曲线的实例代码
Jun 01 #Python
详细介绍python类及类的用法
教你怎么用PyCharm为同一服务器配置多个python解释器
用python批量解压带密码的压缩包
May 31 #Python
You might like
BBS(php & mysql)完整版(七)
2006/10/09 PHP
PHP 变量定义和变量替换的方法
2009/07/30 PHP
php设置页面超时时间解决方法
2015/09/22 PHP
yii用户注册表单验证实例
2015/12/26 PHP
php禁用函数设置及查看方法详解
2016/07/25 PHP
PHP连接sftp并下载文件的方法教程
2018/08/26 PHP
PHP实现获取ip地址的5种方法,以及插入用户登录日志操作示例
2019/02/28 PHP
javascript中input中readonly和disabled区别介绍
2012/10/23 Javascript
js定时器(执行一次、重复执行)
2014/03/07 Javascript
AngularJS + Node.js + MongoDB开发的基于高德地图位置的通讯录
2015/01/02 Javascript
在Node.js应用中读写Redis数据库的简单方法
2015/06/30 Javascript
jQuery实现带延迟效果的滑动菜单代码
2015/09/02 Javascript
Bootstrap3制作搜索框样式的方法
2016/07/11 Javascript
深入理解 JavaScript 中的 JSON
2017/04/06 Javascript
js实现扫雷小程序的示例代码
2017/09/27 Javascript
nodejs实现的连接MySQL数据库功能示例
2018/01/25 NodeJs
vue单页缓存方案分析及实现
2018/09/25 Javascript
单页面vue引入百度统计的使用方法示例详解
2018/10/13 Javascript
angular4自定义组件非input元素实现ngModel双向数据绑定的方法
2018/12/28 Javascript
浅谈关于JS下大批量异步任务按顺序执行解决方案一点思考
2019/01/08 Javascript
在JavaScript中如何访问暂未存在的嵌套对象
2019/06/18 Javascript
初试vue-cli使用HBuilderx打包app的坑
2019/07/17 Javascript
利用python模拟sql语句对员工表格进行增删改查
2017/07/05 Python
用pycharm开发django项目示例代码
2018/10/24 Python
appium+python adb常用命令分享
2020/03/06 Python
python3用urllib抓取贴吧邮箱和QQ实例
2020/03/10 Python
2020新版本pycharm+anaconda+opencv+pyqt环境配置学习笔记,亲测可用
2020/03/24 Python
Python如何对XML 解析
2020/06/28 Python
python爬虫爬取图片的简单代码
2021/01/18 Python
Python爬虫分析微博热搜关键词的实现代码
2021/02/22 Python
德国在线订购鲜花:Fleurop
2018/08/25 全球购物
三陽商会官方网站:Sanyo iStore
2019/05/15 全球购物
我的长生果教学反思
2014/04/28 职场文书
会计工作岗位职责
2015/02/03 职场文书
2015年财务工作总结范文
2015/03/31 职场文书
长江七号观后感
2015/06/11 职场文书