TensorFlow如何指定GPU训练模型
作者:打工人小飞 时间:2022-05-31 10:03:21
如何指定GPU训练模型
Linux 查看当前服务器 GPU 的占用情况可以使用 nvidia-smi 命令,如下所示:
nvidia-smi
关于 nvidia-smi 命令输出的详细解释,可参考笔者的另外一篇文章:GPU状态监测 nvidia-smi 命令详解。
在此不再赘述,本文主要分享一下在用 TensorFlow 训练模型时如何指定 GPU。
在用 TensorFlow 训练深度学习模型的时候,若无手动指定,默认是选用第0块来训练,而且其他几块 GPU 也会被显示占用。
有时候,我们更希望可以自己指定一块或者多块 GPU 来训练模型,接下来介绍一种常用的也是比较有效的方法:
举个例子
如果要指定只用第0块 GPU 训练,可以在 python 代码中如下指定:
import os
os.environ['CUDA_VISIBLE_DEVICES']='0'
运行 python 程序后,可发现就只有第0块 GPU 显示被占用,如上图所示。
如果要指定多块 GPU
可以如下:
os.environ['CUDA_VISIBLE_DEVICES']='1,2'
当然,上述这种方法不太方便,每次运行若换用 GPU 训练都需要改代码,可以在运行 python 程序的时候进行指定:
CUDA_VISIBLE_DEVICES=0 python textCnn.py
来源:https://huangfei.blog.csdn.net/article/details/80292250
标签:TensorFlow,训练模型,GPU


猜你喜欢
win2008下mysql8.0.11升级mysql8.0.17版本详细步骤
2024-01-13 18:04:32

Django auth 应用模块详解
2023-05-20 11:17:17
点选TOP后并不是直接跳到页顶的,而是滚动上去
2023-09-07 02:36:43
javascript canvas实现雨滴效果
2024-04-17 10:02:45

pytorch::Dataloader中的迭代器和生成器应用详解
2021-08-31 03:09:30
Django Rest framework认证组件详细用法
2023-02-20 09:47:08

python列表倒序的几种方法(切片、reverse()、reversed())
2022-01-28 02:46:52
Python Dict找出value大于某值或key大于某值的所有项方式
2023-10-16 19:54:55
使用zabbix监控mongodb的方法
2024-01-21 05:13:16

opencv python图像梯度实例详解
2021-05-17 23:26:30

python使用matplotlib:subplot绘制多个子图的示例
2021-01-26 18:13:08

Python OpenCV实现3种滤镜效果实例
2021-06-04 10:20:27

python @property的用法及含义全面解析
2023-04-06 00:42:28
PHP 截取字符串函数整理(支持gb2312和utf-8)
2024-05-11 10:03:32
解决pycharm安装scrapy DLL load failed:找不到指定的程序的问题
2023-08-29 17:47:32

如何解决pycharm中用matplotlib画图不显示中文的问题
2022-03-18 12:20:27

Golang协程池gopool设计与实现
2024-05-28 15:23:01
python+matplotlib实现动态绘制图片实例代码(交互式绘图)
2022-06-16 15:43:08

PYQT5 vscode联合操作qtdesigner的方法
2023-10-31 16:46:41

详解JS 比较两个Json对象的值是否相等的实例
2024-04-29 13:35:36