TensorFlow如何指定GPU训练模型
作者：打工人小飞 发布时间：2022-05-31 10:03:21
标签：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
来源：https://huangfei.blog.csdn.net/article/details/80292250
0
投稿猜你喜欢
前言gif图就是动态图，它的原理和视频有点类似，也是通过很多静态图片合成的.本篇文章主要介绍，如何利用Python快速合成gif图，主要利用- 特别是linux系统，装了多个python，有时候找不到python的绝对路径，有时候装了个django，又找不到django安装到哪里了。
- <script type="text/javascript"> var params&n
- 想要爬取指定网页中的图片主要需要以下三个步骤：（1）指定网站链接，抓取该网站的源代码（如果使用google浏览器就是按下鼠标右键 ->
- 在python项目中，我们经常会用到lambda,那么lambda是什么呢，有什么作用，下面我们开始介绍1、可以使用lambda关键字创建匿
- 简介在php中，类型的继承使用extends关键字，而且最多只能继承一个父类，php不支持多继承。class MyClass {
- 一、随机数种子为什么要提出随机数种子呢？咱们前面提到过了,随机数均是模拟出来的, 想要模拟的比较真实,就需要变换种子函数内的数值,一般以时间
- 前面我们简单的了解Django的一些工作原理，其中关于页面展示的内容，也全部都是视图（Views）返回的内容，那么我们也知道前端包括很多内容
- 今天在测试php程序的时候，出现了一个错误提示：Cannot use a scalar value as an array，这个错误提示前几
- 磁盘搜索是性能的很大瓶颈。这个问题在数据大量增长以至于无法使用有效的缓存时尤为明显。或多或少随即访问大数据库时，就必然会有至少一次磁盘搜索来
- selenium关闭窗口有两个方法，close与quit，我们稍作研究便知道这两个方法的区别。1.看源码或API这是close()的说明：C
- 介绍本篇文章主要介绍如何爬取麦子学院的课程信息（本爬虫仍是单线程爬虫），在开始介绍之前，先来看看结果示意图怎么样，是不是已经跃跃欲试了？首先
- 使用python实现简单倒计时exe，供大家参考，具体内容如下使用tkinter制作界面实现倒计时功能。使用time.sleep(1)实现
- 什么是python的装饰器？网络上的定义：装饰器就是一函数，用来包装函数的函数，用来修饰原函数，将其重新赋值给原来的标识符，并永久的丧失原函
- 目录前言解决方案工具介绍工作原理使用限制使用注意使用示例部分参数说明输出结果示例腾讯云数据库 MySQL 使用注意总结一下前言写作案例分析，
- 本文实例为大家分享了python学生管理系统开发的具体代码，供大家参考，具体内容如下学生管理系统(基础版)#定义一个函数，显示可以使用的功能
- web跨域请求1.为什么要有跨域限制举个例子：1.用户登录了自己的银行页面 http://mybank.com，http://mybank.
- 本文实例讲述了PHP使用JpGraph绘制折线图操作。分享给大家供大家参考，具体如下：下载jpgraph类库，使用的是src目录下的类文件。
- 一、进程介绍进程：正在执行的程序，由程序、数据和进程控制块组成，是正在执行的程序，程序的一次执行过程，是资源调度的基本单位。程序：没有执行的
- 今天，本文向大家推荐20佳国外的脚本下载网站。1- Hot Scripts2- Code Canyon3- User Scripts4- S
