如何判断pytorch是否支持GPU加速
作者：抬头仰望-y 发布时间：2023-10-06 11:46:49
标签：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()
返回当前设备索引；
来源：https://blog.csdn.net/moshiyaofei/article/details/89703846
0
投稿猜你喜欢
- 本文实例为大家分享了python人民币大小写转换的具体代码，供大家参考，具体内容如下大家应该都知道，银行打印账单有时候会跟上人民币的阿拉伯数
- 一直一来，我都在想办法看能不能用 col 控制表格的列隐藏，但是无奈，FF对COL支持不够好（其实是以前没有发现COL在FF下要怎么弄）。今
- 在Linux上有个常用的命令 curl（非常好用），支持curl的就是大名鼎鼎的libcurl库；libcurl是功能强大的，而且是非常高效
- 近期，MSN、江民等知名网站相继受到了黑客的威胁和攻击，一时间网络上风声鹤唳。本报编辑部接到本文作者（炽天使）的电话，他详细讲述了发现国内最
经常在网站上看到诸如www.abc.com/?news或者www.abc.com/?id=123这样的网址，一开始觉得很神秘，其实现在看多了- 如下所示：import serialimport sysimport osimport timeimport redef wait_for_
- 用了腾讯QQ也有将近十年了，今天心血来潮想模仿腾讯QQ的登陆面板做一个web版的登陆面板，然后参考了一些代码，自己模仿，学写了一个。&nbs
- 记录了Windows安装python3.7的详细过程，供大家参考，具体内容如下1. 在python的官网下载python对应版本：官网地址6
- 本文主要给大家介绍了关于webpack中publicPath路径问题的相关内容，分享出来供大家参考学习，下面来看看详细的介绍：output:
- 开篇：测试过程中，对于多参数参数多值的情况进行测试用例组织，之前一直使用【正交分析法】进行用例组织，说白了就是把每个参数的所有值分别和其他参
- 本文实例为大家分享了python实现图像识别的具体代码，供大家参考，具体内容如下#! /usr/bin/env python from PI
- 前言相信大家在工作无聊时，总想掏出手机，看看微博热搜在讨论什么有趣的话题，但又不方便直接打开微博浏览，今天就和大家分享一个有趣的小爬虫，定时
- 本文实例讲述了微信小程序MUI导航栏透明渐变功能。分享给大家供大家参考，具体如下：导航栏透明渐变效果实现原理1. 利用position:ab
- >>> "hello".encode("hex") '68656c6c6f
- 前言最近公司项目中在使用 Echarts 绘制图表时，由于默认的 label 标签不能满足设计稿需求，所以研究了对 label标签进行格式化
- 问题问题是这样的，要把一个数组存到tfrecord中，然后读取a = np.array([[0, 54, 91, 153, 177,1],
- 本文实例讲述了js实现的全国省市二级联动下拉选择菜单。分享给大家供大家参考。具体如下：运行效果截图如下：具体代码如下：<!DOCTYP
- 前言今天帮师兄赶在deadline之前画论文的图，现学现卖很是刺激，现把使用matplotlib的子库pyplot画折线图和柱状图的代码记录
- 一、制作播放器的思路制作一个多功能音乐播放器的思路确定播放器的需求和功能，例如支持哪些音频格式、播放列表管理、循环播放、暂停、进度条显示等等
- 今天学到Python 有一个名为 random 的内置模块，可用于生成随机数，这个好玩~可以用来做为上课随机点名的程序了哈哈。。def ma
