PyTorch device与cuda.device用法


Posted in Python onApril 03, 2022

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 相关文章推荐
Python中3种内建数据结构:列表、元组和字典
Nov 30 Python
python持久性管理pickle模块详细介绍
Feb 18 Python
Python的Flask框架中实现分页功能的教程
Apr 20 Python
判断网页编码的方法python版
Aug 12 Python
Python微信库:itchat的用法详解
Aug 14 Python
python特性语法之遍历、公共方法、引用
Aug 08 Python
Python正则表达式和re库知识点总结
Feb 11 Python
如何基于python操作excel并获取内容
Dec 24 Python
Pytorch使用MNIST数据集实现基础GAN和DCGAN详解
Jan 10 Python
python绘制玫瑰的实现代码
Mar 02 Python
Python参数传递及收集机制原理解析
Jun 05 Python
使用python脚本自动生成K8S-YAML的方法示例
Jul 12 Python
Python实现为PDF去除水印的示例代码
Apr 03 #Python
Python实现文字pdf转换图片pdf效果
Apr 03 #Python
python 实现图片特效处理
教你使用Python获取QQ音乐某个歌手的歌单
Python os和os.path模块详情
如何通过一篇文章了解Python中的生成器
Python pyecharts绘制条形图详解
You might like
生成sessionid和随机密码的例子
2006/10/09 PHP
PHP新手上路(十二)
2006/10/09 PHP
Alliance vs Liquid BO3 第一场2.13
2021/03/10 DOTA
JavaScript 动态改变图片大小
2009/06/11 Javascript
javascript 读取图片文件的大小
2009/06/25 Javascript
ajax中get和post的说明及使用与区别
2012/12/23 Javascript
jQuery中filter()和find()的区别深入了解
2013/09/25 Javascript
jquery选择器使用详解
2014/04/08 Javascript
如何实现json数据可视化详解
2016/11/24 Javascript
详解nodejs 文本操作模块-fs模块(一)
2016/12/22 NodeJs
jQuery操作json常用方法示例
2017/01/04 Javascript
vue脚手架vue-cli的学习使用教程
2017/06/06 Javascript
微信小程序button组件使用详解
2018/01/31 Javascript
vue中引入第三方字体文件的方法示例
2018/12/17 Javascript
使用webpack搭建vue项目实现脚手架功能
2019/03/15 Javascript
egg.js的基本使用和调用数据库的方法示例
2019/05/18 Javascript
解析原来浏览器原生支持JS Base64编码解码
2019/08/12 Javascript
通过angular CDK实现页面元素拖放的步骤详解
2020/07/01 Javascript
python 合并文件的具体实例
2013/08/08 Python
Python发送Email方法实例
2014/08/21 Python
详解Python实现多进程异步事件驱动引擎
2017/08/25 Python
手把手教你用python抢票回家过年(代码简单)
2018/01/21 Python
对python csv模块配置分隔符和引用符详解
2018/12/12 Python
Python 中的参数传递、返回值、浅拷贝、深拷贝
2019/06/25 Python
python反编译教程之2048小游戏实例
2021/03/03 Python
CSS3改变浏览器滚动条样式
2019/01/04 HTML / CSS
AmazeUI的下载配置与Helloworld的实现
2020/08/19 HTML / CSS
Oracle中delete,truncate和drop的区别
2016/05/05 面试题
金融专业个人求职信范文
2013/11/28 职场文书
建筑工程专业学生的自我评价
2013/12/25 职场文书
运动会通讯稿100字
2014/01/31 职场文书
世界水日宣传活动总结
2015/02/09 职场文书
详解PHP用mb_string处理windows中文字符
2021/05/26 PHP
MySQL中出现乱码问题的终极解决宝典
2021/05/26 MySQL
JavaScript数组reduce()方法的语法与实例解析
2021/07/07 Javascript
python神经网络Xception模型
2022/05/06 Python