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使用内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
string模块主要包含关于字符串的处理函数，这篇文章主要介绍了Python中String模块示例代码，代码简单易懂，对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下2022-12-12
这篇文章主要介绍了python 字典嵌套的相关资料，帮助大家更好的理解和学习python，感兴趣的朋友可以了解下2020-09-09
这篇文章主要为大家介绍了Python中常用的两个序列化模块：pickle序列化和json序列化。文中的示例代码讲解详细，感兴趣的小伙伴可以学习一下2022-10-10
python中parser.add_argument()用法实例(命令行选项、参数和子命令解析器)
最近开始读论文代码了,遇到一个名字叫option的py文件,打开一看清一色的parser.add_argument(),看得是一脸懵逼,这篇文章主要给大家介绍了关于python中parser.add_argument()用法的相关资料,需要的朋友可以参考下2022-03-03
这篇文章主要介绍了如何在django中实现分页功能,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-04-04
这篇文章主要介绍了Python关于抽奖系统的思考与设计思路,本文通过一些简单的例子来说一说抽奖系统背后的逻辑,看看究竟是你运气不好还是系统逻辑在作怪,需要的朋友可以参考下2023-03-03
这篇文章主要介绍了pycharm中配置当鼠标悬停时快速提示方法参数,本文图文并茂给大家介绍的非常详细，具有一定的参考借鉴价值，需要的朋友可以参考下2019-07-07
这篇文章主要介绍了Python3+Flask安装使用教程详解,需要的朋友可以参考下2021-02-02
这篇文章主要介绍了python中cPickle用法例子，有需要的朋友可以参考一下2014-01-01
今天小编就为大家分享一篇浅谈pytorch和Numpy的区别以及相互转换方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-07-07
最新评论