PyTorch device与cuda.device用法介绍

更新时间：2022年04月02日 16:03:47 作者：研究生不迟到

这篇文章主要介绍了PyTorch device与cuda.device用法,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下

1 查看当前的device

输入情况：

import torch
print("Default Device : {}".format(torch.Tensor([4, 5, 6]).device))

输出情况：

Default Device : cpu

2 cpu设备可以使用“cpu:0”来指定

输入情况

device = torch.Tensor([1, 2, 3], device="cpu:0").device
print("Device Type: {}".format(device))

输出情况

Device Type: cpu

3 gpu设备可以使用“cuda:0”来指定

输入情况

gpu = torch.device("cuda:0")
print("GPU Device:【{}：{}】".format(gpu.type, gpu.index))

输出情况

GPU Device:【cuda：0】

4 查询CPU和GPU设备数量

输入情况

print("Total GPU Count :{}".format(torch.cuda.device_count()))
print("Total CPU Count :{}".format(torch.cuda.os.cpu_count()))

输出情况

Total GPU Count :1
Total CPU Count :8

5 从CPU设备上转换到GPU设备

5.1 torch.Tensor方法默认使用CPU设备

输入情况

data = torch.Tensor([[1, 4, 7], [3, 6, 9], [2, 5, 8]])
print(data.shape)

输出情况

torch.Size([3, 3])

5.2 使用to方法将cpu的Tensor转换到GPU设备上

输入情况：

data_gpu = data.to(torch.device("cuda:0"))
print(data_gpu.device)

输出情况：

cuda:0

5.3 使用.cuda方法将cpu的Tensor转换到GPU设备上

输入情况：

data_gpu2 = data.cuda(torch.device("cuda:0"))
# 如果只有一块gpu的话  直接写成这样：data_gpu2 = data.cuda()
print(data_gpu2.device)

输出情况：

cuda:0

到此这篇关于PyTorch device与cuda.device用法的文章就介绍到这了,更多相关PyTorch device使用内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

python使用glob检索文件的操作
这篇文章主要介绍了python使用glob检索文件的操作，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
2021-05-05
python人工智能tensorflow函数tf.nn.dropout使用方法
这篇文章主要为大家介绍了python人工智能tensorflow函数tf.nn.dropout使用示例，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
2022-05-05
Python wxPython库使用wx.ListBox创建列表框示例
这篇文章主要介绍了Python wxPython库使用wx.ListBox创建列表框,结合实例形式分析了wxPython库使用wx.ListBox创建列表框的简单实现方法及ListBox函数相关选项的功能,需要的朋友可以参考下
2018-09-09
python使用pip安装模块出现ReadTimeoutError: HTTPSConnectionPool的解决方法
这篇文章主要介绍了python使用pip安装模块出现ReadTimeoutError: HTTPSConnectionPool的解决方法,需要的朋友可以参考下
2019-10-10
使用Python来开发Markdown脚本扩展的实例分享
这篇文章主要介绍了使用Python来开发Markdown脚本扩展的实例分享,文中的示例是用来简单地转换文档结构,主要为了体现一个思路,需要的朋友可以参考下
2016-03-03
Python操作MongoDB数据库PyMongo库使用方法
这篇文章主要介绍了Python操作MongoDB数据库PyMongo库使用方法,本文讲解了创建连接、连接数据库、连接聚集、查看全部聚集名称、查看聚集的一条记录等操作方法,需要的朋友可以参考下
2015-04-04
python解析PDF程序代码
python解析pdf信息，目前还是不能做到很完善，每个方法都会有它的弊端，都需要结合自己的业务场景来和其它方法一块进行使用，从而达到一个相对满意的目标，接下来给大家介绍python解析PDF的程序代码，一起看看吧
2021-06-06
python机器学习库常用汇总
这篇文章主要为大家汇总了常用python机器学习库，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2017-11-11
vscode+PyQt5安装详解步骤
这篇文章主要介绍了vscode+PyQt5安装详解步骤,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2020-08-08
python计算两个数的百分比方法
今天小编就为大家分享一篇python计算两个数的百分比方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2018-06-06