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调用C语言的方法【基于ctypes模块】
Jan 22 Python
Python统计一个字符串中每个字符出现了多少次的方法【字符串转换为列表再统计】
May 05 Python
python django下载大的csv文件实现方法分析
Jul 19 Python
Pytorch实现基于CharRNN的文本分类与生成示例
Jan 08 Python
开启Django博客的RSS功能的实现方法
Feb 17 Python
python实现字符串和数字拼接
Mar 02 Python
Python 读取WAV音频文件 画频谱的实例
Mar 14 Python
python实现超级玛丽游戏
Mar 18 Python
Django Xadmin多对多字段过滤实例
Apr 07 Python
学习python需要有编程基础吗
Jun 02 Python
Python3如何使用range函数替代xrange函数
Oct 05 Python
python爬取微博评论的实例讲解
Jan 15 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
php自动适应范围的分页代码
2008/08/05 PHP
PHP 读取文件的正确方法
2009/04/29 PHP
解析zend studio中直接导入svn中的项目的方法步骤
2013/06/21 PHP
php获取本周星期一具体日期的方法
2015/04/20 PHP
PHP实现上传多文件示例代码
2017/02/20 PHP
JavaScript中的对象化编程
2008/01/16 Javascript
替代window.event.srcElement效果的可兼容性的函数
2009/12/18 Javascript
JavaScript XML和string相互转化实现代码
2011/07/04 Javascript
JavaScript建立一个语法高亮输入框实现思路
2013/02/26 Javascript
JavaScript indexOf方法入门实例(计算指定字符在字符串中首次出现的位置)
2014/10/17 Javascript
解决angular的post请求后SpringMVC后台接收不到参数值问题的方法
2015/12/10 Javascript
Express的路由详解
2015/12/10 Javascript
浅析AngularJS Filter用法
2015/12/28 Javascript
逻辑表达式中与或非的用法详解
2016/06/06 Javascript
使用JavaScript获取Request中参数的值方法
2016/09/27 Javascript
Javascript将字符串日期格式化为yyyy-mm-dd的方法
2016/10/27 Javascript
Angular使用ng-messages与PHP进行表单数据验证
2016/12/28 Javascript
通过BootStrap-select插件 js jQuery控制select属性变化
2017/01/03 Javascript
JS实现本地存储信息的方法(基于localStorage与userData)
2017/02/18 Javascript
Vue一个案例引发的递归组件的使用详解
2018/11/15 Javascript
Vue插槽原理与用法详解
2019/03/05 Javascript
vue cli 3.x 项目部署到 github pages的方法
2019/04/17 Javascript
微信小程序自定义tabBar在uni-app的适配详解
2019/09/30 Javascript
浅谈vue生命周期共有几个阶段?分别是什么?
2020/08/07 Javascript
Python数据结构与算法之链表定义与用法实例详解【单链表、循环链表】
2017/09/28 Python
Pycharm设置界面全黑的方法
2018/05/23 Python
python识别图像并提取文字的实现方法
2019/06/28 Python
python挖矿算力测试程序详解
2019/07/03 Python
python实现操作文件(文件夹)
2019/10/31 Python
Python类的绑定方法和非绑定方法实例解析
2020/03/04 Python
python名片管理系统开发
2020/06/18 Python
python实现邮件循环自动发件功能
2020/09/11 Python
Vector, ArrayList, HashTable, HashMap哪些是线程安全的,哪些不是
2015/10/12 面试题
电焊工岗位工作职责
2014/07/09 职场文书
领导干部群众路线教育实践活动剖析材料
2014/10/10 职场文书
2019奶茶店创业计划书范本,值得你借鉴
2019/08/14 职场文书