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使用PyV8执行javascript代码示例分享
Dec 04 Python
zbar解码二维码和条形码示例
Feb 07 Python
Python的网络编程库Gevent的安装及使用技巧
Jun 24 Python
python实现读Excel写入.txt的方法
Apr 29 Python
pandas 条件搜索返回列表的方法
Oct 30 Python
Python进程间通信 multiProcessing Queue队列实现详解
Sep 23 Python
Python中BeautifuSoup库的用法使用详解
Nov 15 Python
Python函数参数定义及传递方式解析
Jun 10 Python
使用python修改文件并立即写回到原始位置操作(inplace读写)
Jun 28 Python
Windows下PyCharm配置Anaconda环境(超详细教程)
Jul 31 Python
Python实现播放和录制声音的功能
Aug 12 Python
python代码实现猜拳小游戏
Nov 30 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
windows环境下php配置memcache的具体操作步骤
2013/06/09 PHP
PHP实现获取文件后缀名的几种常用方法
2015/08/08 PHP
php app支付宝回调(异步通知)详解
2018/07/25 PHP
Javascript 代码也可以变得优美的实现方法
2009/06/22 Javascript
模仿JQuery.extend函数扩展自己对象的js代码
2009/12/09 Javascript
input链接页面、打开新网页等等的具体实现
2013/12/30 Javascript
jquery 遍历数组 each 方法详解
2016/05/25 Javascript
js仿支付宝多方框输入支付密码效果
2016/09/27 Javascript
jquery中用函数来设置css样式
2016/12/22 Javascript
jquery.tableSort.js表格排序插件使用方法详解
2020/08/12 Javascript
Vue-cropper 图片裁剪的基本原理及思路讲解
2018/04/17 Javascript
node使用Mongoose类库实现简单的增删改查
2018/11/08 Javascript
详解nuxt 微信公众号支付遇到的问题与解决
2019/08/26 Javascript
多个vue子路由文件自动化合并的方法
2019/09/03 Javascript
JS实现随机点名器
2020/04/12 Javascript
Element实现表格嵌套、多个表格共用一个表头的方法
2020/05/09 Javascript
Vue 中如何将函数作为 props 传递给组件的实现代码
2020/05/12 Javascript
vue页面引入three.js实现3d动画场景操作
2020/08/10 Javascript
利用python生成一个导出数据库的bat脚本文件的方法
2016/12/30 Python
Python绘制七段数码管实例代码
2017/12/20 Python
pandas修改DataFrame列名的实现方法
2019/02/22 Python
GitHub 热门:Python 算法大全,Star 超过 2 万
2019/04/29 Python
Python3+OpenCV2实现图像的几何变换(平移、镜像、缩放、旋转、仿射)
2019/05/13 Python
Python常用扩展插件使用教程解析
2020/11/02 Python
python 日志模块logging的使用场景及示例
2021/01/04 Python
纯CSS3实现手风琴风格菜单具体步骤
2013/05/06 HTML / CSS
关于前端上传文件全面基础扫盲贴(入门)
2019/08/01 HTML / CSS
使用layui实现左侧菜单栏及动态操作tab项的方法
2020/11/10 HTML / CSS
JoJo Maman Bébé爱尔兰官网:英国最受欢迎的精品母婴品牌
2020/12/20 全球购物
Ibatis中如何提高SQL Map的性能
2013/05/11 面试题
商务助理岗位职责
2013/11/13 职场文书
委托书样本
2014/04/02 职场文书
狮子林导游词
2015/02/03 职场文书
2015年派出所民警工作总结
2015/04/24 职场文书
2015年学校保卫部工作总结
2015/05/11 职场文书
2015年公司保安年终工作总结
2015/05/14 职场文书