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

更新时间：2021年06月01日 09:40:34 作者：抬头仰望-y

这篇文章主要介绍了判断pytorch是否支持GPU加速的操作，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教

加上这句代码：

print torch.cuda.is_available()

判断完毕！说说在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()

返回当前设备索引；

在这里插入图片描述

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

Python中带时区的日期转换工具类总结
这篇文章主要为大家详细介绍了一些Python中带时区的日期转换工具类，文中的示例代码讲解详细，具有一定的学习价值，感兴趣的小伙伴可以跟随小编一起了解一下
2023-05-05
OpenCV实现背景分离(证件照背景替换)
本文主要介绍了OpenCV实现背景分离(证件照背景替换)，文中通过示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2021-09-09
python去除字符串中的换行符
这篇文章主要介绍了python去除字符串中的换行符的相关资料，然后在文章下面给大家补充介绍了python去除空格和换行符的方法,需要的朋友可以参考下
2017-10-10
使用Python生成你的LaTeX公式基础使用
这篇文章主要介绍了使用Python生成你的LaTeX公式基础使用,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
2024-01-01
Django nginx配置实现过程详解
这篇文章主要介绍了Django nginx配置实现过程详解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2020-09-09
python实现pptx批量向PPT中插入图片
大家好，本篇文章主要讲的是python实现pptx批量向PPT中插入图片，感兴趣的同学赶快来看一看吧，对你有帮助的话记得收藏一下
2022-02-02
Python按行读取文件的简单实现方法
下面小编就为大家带来一篇Python按行读取文件的简单实现方法。小编觉得挺不错的，现在就分享给大家，也给大家做个参考。一起跟随小编过来看看吧
2016-06-06
学习python的前途 python挣钱
在本文里我们给大家整理了关于学习python的前途以及python挣钱的方法整理，有兴趣的朋友们阅读下。
2019-02-02
K最近邻算法(KNN)---sklearn+python实现方式
今天小编就为大家分享一篇K最近邻算法(KNN)---sklearn+python实现方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-02-02
Python新手在作用域方面经常容易碰到的问题
这篇文章主要介绍了Python新手在作用域方面经常容易碰到的问题,全局变量和局部变量方面的知识在Python学习当中是基础中的基础,需要的朋友可以参考下
2015-04-04