Pytorch中GPU使用
本文中的内容转载自:《动手学深度学习》
GPU计算
到目前为止,我们一直在使用CPU计算。对复杂的神经网络和大规模的数据来说,使用CPU来计算可能不够高效。在本节中,我们将介绍如何使用单块NVIDIA GPU来计算。所以需要确保已经安装好了PyTorch GPU版本。准备工作都完成后,下面就可以通过nvidia-smi
命令来查看显卡信息了。
!nvidia-smi # 对Linux/macOS用户有效
输出:
Sun Mar 17 14:59:57 2019
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 390.48 Driver Version: 390.48 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|===============================+======================+======================|
| 0 GeForce GTX 1050 Off | 00000000:01:00.0 Off | N/A |
| 20% 36C P5 N/A / 75W | 1223MiB / 2000MiB | 0% Default |
+-------------------------------+----------------------+----------------------++-----------------------------------------------------------------------------+
| Processes: GPU Memory |
| GPU PID Type Process name Usage |
|=============================================================================|
| 0 1235 G /usr/lib/xorg/Xorg 434MiB |
| 0 2095 G compiz 163MiB |
| 0 2660 G /opt/teamviewer/tv_bin/TeamViewer 5MiB |
| 0 4166 G /proc/self/exe 416MiB |
| 0 13274 C /home/tss/anaconda3/bin/python 191MiB |
+-----------------------------------------------------------------------------+
可以看到我这里只有一块GTX 1050,显存一共只有2000M(太惨了
Pytorch中GPU使用相关推荐
- Pytorch中多GPU训练指北
前言 在数据越来越多的时代,随着模型规模参数的增多,以及数据量的不断提升,使用多GPU去训练是不可避免的事情.Pytorch在0.4.0及以后的版本中已经提供了多GPU训练的方式,本文简单讲解下使用P ...
- gpu处理信号_在PyTorch中使用DistributedDataParallel进行多GPU分布式模型训练
先进的深度学习模型参数正以指数级速度增长:去年的GPT-2有大约7.5亿个参数,今年的GPT-3有1750亿个参数.虽然GPT是一个比较极端的例子但是各种SOTA模型正在推动越来越大的模型进入生产应用 ...
- 深度学习PyTorch,TensorFlow中GPU利用率较低,使用率周期性变化的问题
在用tensorflow训练神经网络时,发现训练迭代的速度时而快时而慢,监督的GPU使用率也是周期性变化,通过了解,发现原因是: GPU在等待CPU读取,预处理,并传输数据过来,因此要提高GPU的使用 ...
- Anaconda中GPU版本Pytorch 的whl 安装方法【2023.1最新最详细】(附anaconda以及cudacudnn安装教程)
本教程郑重承诺,不使用Pytorch官网,不使用清华源镜像资源,全部利用whl文件完成Pytorch 安装. 写在前面的一些艰辛心路历程: 因为毕设需要安装pytorch,学长轻飘飘地丢给我一句&qu ...
- anaconda中使用pip安装pytorch(GPU)
anaconda中使用pip安装pytorch(GPU) 由于conda直接安装torch遇到了问题,在此记录一下另一种装法. 一.确认cuda与cudnn安装完成 我在以前已经安装完成了 在这仅确认 ...
- 在Pytorch中使用GPU加速模型训练
CPU->GPU 当我们需要训练一个复杂模型的时候,使用CPU来进行训练往往会花费非常多的时间.因此,一般情况下,我们都会选择在GPU (Graphics Process Unit)环境下进行训 ...
- pytorch 多GPU训练总结(DataParallel的使用)
版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明. 本文链接:https://blog.csdn.net/weixin_40087578/arti ...
- 利用 AssemblyAI 在 PyTorch 中建立端到端的语音识别模型
作者 | Comet 译者 | 天道酬勤,责编 | Carol 出品 | AI 科技大本营(ID:rgznai100) 这篇文章是由AssemblyAI的机器学习研究工程师Michael Nguyen ...
- 实践教程 | 浅谈 PyTorch 中的 tensor 及使用
点击上方"视学算法",选择加"星标"或"置顶" 重磅干货,第一时间送达 作者 | xiaopl@知乎(已授权) 来源 | https://z ...
最新文章
- Spring Cloud Alibaba 之 服务治理:Nacos 如何实现微服务服务治理
- 浅谈代码的执行效率(4):汇编优化
- postgres 备份 恢复
- iOS开发系列--无限循环的图片浏览器
- Linux驱动之Input子系统要点分析
- 芋道 Spring Boot 自动配置原理
- 字符,字符串,int之间互相转换
- 舰r4月28服务器维护,崩坏3 11月28日版本更新维护通知
- mysql 多个值求和_SQL优化大神玩转MySQL函数系列(2)LEAST,SUM的应用
- reticulate: R interface to Python
- 拓端tecdat|R语言在BRFSS数据中可视化分析探索糖尿病的影响因素
- 5脚耳机插座原理图_手机原理图的解读
- 计算机代数与密码学,代数曲线与密码学
- 用Python怎样实现Excel中的公式向下填充?
- Jeff Dean:谷歌大脑背后的“大脑” | AI名人堂
- 外贸软件出口管理系统亮点及重点
- web中html+CSS修改背景图片的不透明度
- Python 进阶视频课 - 14. FR007 利率掉期定价和曲线拔靴
- Spring-Aop及jdbc总结
- ThinkPad笔记本如何重装系统
热门文章
- 2015年《大数据》高被引论文Top10文章No.2——大数据时代的数据挖掘 —— 从应用的角度看大数据挖掘(上)...
- 【面向对象】对象的本质
- 【项目管理】CMM能力成熟度模型
- 【面向对象】面向对象程序设计测试题13-Java异常处理测试题
- 【数据结构与算法】链式栈的Java实现
- win7下cocos2dx2.2+vs2010+python2.7环境搭建
- MySQL数据库如何做好优化
- 跳转到系统默认的Home
- 企业应用架构模式笔记
- [数据库字典]简单的3个SQL视图搞定所有SqlServer数据库字典