PyTorch device与cuda.device用法介绍
作者：研究生不迟到 发布时间：2022-07-26 23:49:10
标签：PyTorch,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
来源：https://blog.csdn.net/weixin_42521185/article/details/123913221
0
投稿猜你喜欢
- numpy中的ndarray转化成pytorch中的tensor : torch.from_numpy()pytorch中的tensor转化
大家好，我是辣条。曾经有一个真挚的机会，摆在我面前，但是我没有珍惜，等到失去的时候才后悔莫及，尘世间最痛苦的事莫过于此，如果老天可以再给我一- 在Web上使用菜单可以极大地节约页面的空间，同时也比较的符合用户从Windows上继承下来的UI操作体验。在以往的Web页菜单设计中，我们普
- php有哪些优点？PHP优点：1.入门快，有其它语言基础的程序员二周左右的时间就可以入门，一个月左右的时间基本上就可以开发简单的项目了。2.
- Google Chrome，又喜欢逛豆瓣，Google Chrome支持User Scripts了，但是我写的那几个插件在Chrome下又跑
- Python字符串字符串或串(String)是由数字、字母、下划线组成的一串字符。一般记为 :s="a1a2···an"
- 通过前面内容的介绍，我们对 Surface 对象有了大体上的认识。Pygame 针对文本、图像、颜色提供了不同模块来生成它们各自的 Surf
- Python request获取网页中文乱码问题r = requests.get(“http://www.baidu.com“)
- 前言有时候在使用Python处理比较耗时操作的时候，为了便于观察处理进度，这时候就需要通过进度条将处理情况进行可视化展示，以便我们能够及时了
- 本文实例讲述了python实现的自动发送消息功能。分享给大家供大家参考，具体如下：一个简单的脚本#-*- coding:utf-8 -*-f
- set global log_bin_trust_function_creators = 1; DROP FUNCTION IF EXIST
- 1.前言Python中函数的参数类型比较丰富，比如我们经常见到*args和**kwargs作为参数。初学者遇到这个多少都有点懵逼，今天我们来
- 1. 简介在windows系统上，重复性的操作可以用Python脚本来完成，其中常用的模块是win32gui、win32con、win32a
- 占位符通过占位符，可以指定格式进行输入或输出，以下为 fmt 标准库里的占位符：普通占位符占位符描述举例结果％v默认格式的值fmt.Prin
- 腾讯“月捐计划”倡导爱心人士，通过每月小额捐款的形式，长期关注和支持公益项目。并和亿万爱心网友一起，每人每月1份爱，点滴付出，汇成爱海，形成
- 本文实例讲述了Python zip()函数用法。分享给大家供大家参考，具体如下：这里介绍python中zip()函数的使用：>>
- 预测结果转为numpy：logits=model(feature)#如果模型是跑在GPU上result=logits.data.cpu().
- 一、意义：当我们使用一个数据库时，总希望数据库的内容是可靠的、正确的，但由于计算机系统的故障(硬件故障、网络故障、进程故障和系统故障)影响数
- 阅读上一篇：浏览器中的内存泄露 4.内存泄露的解决方案显式类型转换 首先说说最容易处理的情况 对于类型转换造成的错误，我们可以通过显式类型转
- Go对字符串格式化提供了良好的支持。下面我们看些常用的字符串格式化的例子。package mainimport "fmt"
