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针对谷歌音乐相关信息的获取技巧,具有一定参考借鉴价值,需要的朋友可以参考下2015-07-07
这篇文章主要介绍了django跳转页面传参的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-09-09
广告点击率是指有多少用户点击了您的广告与有多少用户查看了您的广告的比率,本文主要为大家介绍了如何使用Python实现广告点击率预测,感兴趣的小伙伴可以了解下2023-10-10
这篇文章主要介绍了Python实现自定义读写分离代码实例,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-11-11
在本篇文章里小编给大家整理的是一篇关于Python-split()函数实例用法讲解，有兴趣的朋友们可以学习下。2020-12-12
这篇文章主要给大家介绍了关于Python数据处理利器Slice函数用法的相关资料,slice函数是Python中的一个内置函数,用于对序列进行切片操作,文中通过代码介绍的非常详细,需要的朋友可以参考下2024-03-03
Python实现的对本地host127.0.0.1主机进行扫描端口功能示例
这篇文章主要介绍了Python实现的对本地host127.0.0.1主机进行扫描端口功能,可实现扫描本机开放端口的功能,涉及Python socket模块与Thread多线程模块相关使用技巧,需要的朋友可以参考下2019-02-02
这篇文章主要介绍了Python 开发中的高级技巧，非常不错，具有收藏价值，感兴趣的朋友一起看看吧2018-11-11
本文主要介绍了numpy稀疏矩阵的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2022-08-08
层次聚类( Hierarchical Clustering )是聚类算法的一种,通过计算不同类别的相似度类创建一个有层次的嵌套的树,下面这篇文章主要给大家介绍了关于Python中层次聚类的详细讲解,需要的朋友可以参考下2022-12-12
最新评论