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使用openpyxl实现对excel表格相对路径的超链接的创建方式
这篇文章主要介绍了python使用openpyxl实现对excel表格相对路径的超链接的创建方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教2024-03-03python判断列表字典字符串元组是否存在某个值或者空值(多种方法)
这篇文章主要介绍了python判断列表字典字符串元组是否存在某个值或者空值,本文通过实例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友参考下吧2024-02-02Python第三方库qrcode或MyQr生成博客地址二维码
使用第三方库qrcode或者MyQr给自己的博客网址快速生成二维码,一键分享,文中含有详细示例代码,有需要的朋友可以借鉴参考下,希望能够有所帮助2021-10-10
最新评论