TensorFlow如何指定GPU训练模型
更新时间：2022年11月02日 14:53:44 作者：打工人小飞
这篇文章主要介绍了TensorFlow如何指定GPU训练模型，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
如何指定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
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
在Python中,复合型数据（例如列表、元组、集合和字典）是非常常用的数据类型,本文将深入探讨Python复合型数据的常见陷阱,并提供一些避免这些问题的实用建议和技巧,希望对大家有所帮助2024-03-03
Python中布尔值(Booleans)表示以下两个值之一：True或False。本文主要介绍布尔值(Booleans)的使用，和使用时需要注意的地方，需要的可以参考一下2023-02-02
python3中利用filter函数输出小于某个数的所有回文数实例
今天小编就为大家分享一篇 python3中利用filter函数输出小于某个数的所有回文数实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-11-11
模板是 Django MVT 结构的第三个也是最重要的部分。Django中的模板基本上是用HTML，CSS和Javascript编写在.html文件中的。本文将通过实例详细聊聊Django模板的使用，感兴趣的可以2022-10-10
今天小编就为大家分享一篇pytorch获取vgg16-feature层输出的例子，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-08-08
python项目中requirements.txt的用法实例教程
Python项目中必须包含一个requirements.txt文件,用于记录所有依赖包及其精确的版本号,以便新环境部署,下面这篇文章主要给大家介绍了关于python项目中requirements.txt用法的相关资料,需要的朋友可以参考下2022-06-06
这篇文章主要介绍了python实现的DES加密算法和3DES加密算法,以实例形式较为详细的分析了DES加密算法和3DES加密算法的原理与实现技巧,需要的朋友可以参考下2015-06-06
这篇文章主要介绍了Python装饰器原理与简单用法,结合实例形式分析了Python装饰器的概念、原理、使用方法及相关注意事项,需要的朋友可以参考下2018-04-04
这篇文章主要介绍了在Python的Django框架中为代码添加注释的方法,需要的朋友可以参考下2015-07-07
这篇文章主要介绍了用Python编写一个基于终端的实现翻译的脚本,代码基于Python2.x,需要的朋友可以参考下2015-04-04
最新评论