如何判断pytorch是否支持GPU加速
作者:抬头仰望-y 时间:2023-10-06 11:46:49
加上这句代码:
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
标签:pytorch,GPU,加速
![](/images/zang.png)
![](/images/jiucuo.png)
猜你喜欢
Python3.10的一些新特性原理分析
2023-06-17 06:35:10
![](https://img.aspxhome.com/file/2023/0/134580_0s.jpg)
SQL Server 2019自定义安装教程
2024-01-12 21:14:22
![](https://img.aspxhome.com/file/2023/4/94434_0s.jpg)
python 字典的概念叙述和使用方法
2021-11-08 00:42:06
![](https://img.aspxhome.com/file/2023/2/114482_0s.png)
Mysql的语句生成后门木马的方法
2024-01-15 20:46:23
go语言睡眠排序算法实例分析
2023-07-15 17:42:56
Windows10下安装配置 perl 环境的详细教程
2022-11-17 01:46:32
![](https://img.aspxhome.com/file/2023/0/65830_0s.jpg)
webpack5搭建一个简易的react脚手架项目实践
2024-04-18 10:02:37
![](https://img.aspxhome.com/file/2023/6/136136_0s.jpg)
Django集成CAS单点登录的方法示例
2023-02-20 14:51:09
![](https://img.aspxhome.com/file/2023/8/70048_0s.jpg)
Golang 分割字符串的实现示例
2024-02-23 03:26:10
opencv+python实现图像矫正
2022-09-28 09:28:14
![](https://img.aspxhome.com/file/2023/5/78815_0s.jpg)
详解python程序中的多任务
2021-01-10 18:55:37
![](https://img.aspxhome.com/file/2023/6/68876_0s.gif)
go语言实现并发网络爬虫的示例代码
2024-01-31 07:45:35
![](https://img.aspxhome.com/file/2023/5/109715_0s.png)
Python计算三角函数之asin()方法的使用
2023-08-04 22:31:46
用pandas中的DataFrame时选取行或列的方法
2023-05-15 07:43:04
Python 对象序列化与反序列化之pickle json详细解析
2021-09-06 23:44:06
OpenCV 边缘检测
2023-08-19 22:54:20
![](https://img.aspxhome.com/file/2023/0/92120_0s.jpg)
Python实现视频裁剪的示例代码
2022-07-20 07:14:14
![](https://img.aspxhome.com/file/2023/3/85373_0s.png)
mysql5.7.10开启慢查询详解
2024-01-18 10:23:23
python之DataFrame实现excel合并单元格
2021-02-22 04:45:00
![](https://img.aspxhome.com/file/2023/7/109557_0s.png)
Python中操作符重载用法分析
2023-04-05 12:34:51