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显示天气预报
Mar 02 Python
Python ORM框架SQLAlchemy学习笔记之关系映射实例
Jun 10 Python
Python实现的简单算术游戏实例
May 26 Python
详解C++编程中一元运算符的重载
Jan 19 Python
关于python pyqt5安装失败问题的解决方法
Aug 08 Python
Django框架封装外部函数示例
May 28 Python
wxPython实现画图板
Aug 27 Python
基于Python和PyYAML读取yaml配置文件数据
Jan 13 Python
如何配置关联Python 解释器 Anaconda的教程(图解)
Apr 30 Python
利用python对excel中一列的时间数据更改格式操作
Jul 14 Python
python基础之匿名函数详解
Apr 21 Python
Python数据可视化之基于pyecharts实现的地理图表的绘制
Jun 10 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
Codeigniter实现多文件上传并创建多个缩略图
2014/06/12 PHP
php构造函数与析构函数
2016/04/23 PHP
php生成毫秒时间戳的实例讲解
2017/09/22 PHP
jquery 获取自定义属性(attr和prop)的实现代码
2012/06/27 Javascript
整理Javascript数组学习笔记
2015/11/29 Javascript
JS 实现可停顿的垂直滚动实例代码
2016/11/23 Javascript
详解用vue-cli来搭建vue项目和webpack
2017/04/20 Javascript
vue实现验证码按钮倒计时功能
2018/04/10 Javascript
JS验证输入的是否是数字及保留几位小数问题
2018/05/09 Javascript
Nodejs中使用puppeteer控制浏览器中视频播放功能
2019/08/26 NodeJs
三步实现ionic3点击退出app程序
2019/09/17 Javascript
微信小程序自定义组件components(代码详解)
2019/10/21 Javascript
JS co 函数库的含义和用法实例总结
2020/04/08 Javascript
VUE中V-IF条件判断改变元素的样式操作
2020/08/09 Javascript
[46:43]DOTA2上海特级锦标赛D组小组赛#1 EG VS COL第三局
2016/02/28 DOTA
python求素数示例分享
2014/02/16 Python
python3利用smtplib通过qq邮箱发送邮件方法示例
2017/12/03 Python
python的numpy模块安装不成功简单解决方法总结
2017/12/23 Python
python tkinter界面居中显示的方法
2018/10/11 Python
Python学习笔记之For循环用法详解
2019/08/14 Python
Python 切分数组实例解析
2019/11/07 Python
如何基于python操作excel并获取内容
2019/12/24 Python
Django实现celery定时任务过程解析
2020/04/21 Python
Python 处理日期时间的Arrow库使用
2020/08/18 Python
英国皇室御用百货:福南梅森(Fortnum & Mason)
2017/12/03 全球购物
英国和国际包裹递送:ParcelCompare
2019/08/26 全球购物
简述网络文件系统NFS,并说明其作用
2016/10/19 面试题
师生聚会感言
2014/01/26 职场文书
年会主持词结束语
2014/03/27 职场文书
计算机网络专业求职信
2014/06/05 职场文书
物流专业求职信
2014/06/30 职场文书
乡镇干部个人对照检查材料思想汇报
2014/10/04 职场文书
反对四风自我剖析材料
2014/10/07 职场文书
优秀团员个人总结
2015/02/26 职场文书
Python中常见的导入方式总结
2021/05/06 Python
全面盘点MySQL中的那些重要日志文件
2021/11/27 MySQL