tensorflow指定GPU与动态分配GPU memory设置
作者：sherry颖 发布时间：2023-01-16 11:30:24
标签：tensorflow,GPU,分配,memory
在tensorflow中，默认指定占用所有的GPU，如需指定占用的GPU，可以在命令行中：
export CUDA_VISIBLE_DEVICES=1
这样便是只占用1号GPU，通过命令
nvidia-smi
可以查看各个GPU的使用情况。
另外，也可以在python程序中指定GPU，并且动态分配memory，代码如下
import os
import sys
os.environ['CUDA_VISIBLE_DEVICES'] = sys.argv[1]
import tensorflow as tf
from keras.backend.tensorflow_backend import set_session
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
set_session(tf.Session(config=config))
这样，占用的GPU就是sys.argv[1]了，并且会按需分配memory。
来源：https://blog.csdn.net/sherry_jy/article/details/79564930
0
投稿猜你喜欢
- Python单元测试unittest中提供了一下四种装饰器实现测试跳过和预期故障。（使用Python 2.7.13）请查考Python手册中
- 链表链表(linked list)是由一组被称为结点的数据元素组成的数据结构，每个结点都包含结点本身的信息和指向下一个结点的地址。由于每个结
前言数据集下载地址：链接: https://pan.baidu.com/s/17aglKyKFvMvcug0xrOqJdQ?pwd=6i7m- 简介pandas中的DF数据类型可以像数据库表格一样进行groupby操作。通常来说groupby操作可以分为三部分：分割数据，应用变换和和
- torch.save：保存序列化的对象到磁盘，使用了Python的pickle进行序列化，模型、张量、所有对象的字典。torch.load：
- 这篇文章主要是想说，“引用只能指向具体对象而不能指向引用”//创建变量testArray 并引用数组 ["1&
- <％ String st = ""; for(int i = 1; i <= 9;
- 本文实例为大家分享了Django实现分页功能，为了容易区别功能的展现，先创建一个数据库，用数据库中的数据做演示。创建数据库步骤如下：1.创建
- UCD介绍UCD是Unicode字符数据库（Unicode Character DataBase）的缩写。UCD由一些描述Unicode字符
- JavaScript游戏开发之键盘控制层的移动截图：<html> <head> <meta http-equi
- MYSQL数据库以它短小、方便、速度快、免费等优点成为很多网站目前首选数据库，但一般都是用PHP+MYSQL相结合来开发各种动态页面，其实A
- 一个小代码 类似资源管理器效果 支持鼠标拖拽 没写些细节东西 JavaScript语言:
- <?php $monthoneday=date("Ym")."01"; $oneweekday
- 之前在《首都机场的点烟器》中分析了一个软件系统所处的状态并且列举了不同的状态所需要的展示给用户的各类信息，我们先简单回顾一下：要设计一个软件
- 我们可以通过更改注册表来解决这个问题： 打开注册表，进入到[HKEY_USERS\.DEFAULT\Control Panel\Intern
- 前言：本文介绍如何构建简单线性回归模型及计算其准确率，最后介绍如何持久化模型。线性回归模型线性回归表示发现函数使用线性组合表示输入变量。简单
- 最近各地中小学都在开展线上教学，有些不自觉的小朋友们用电脑在线学习的时候会趁家长不在的时候偷偷玩游戏、看漫画。本程序screenshot.p
- 恭喜您，您中奖了，你的中奖码是（请牢记，领奖需要）：ＸＸＸＸＸＸＸＸＸＸＸ然后用户输入ＸＸＸＸＸＸＸＸＸＸＸ，简单验证后就可以领奖了。你使用
- 线程进程和线程什么是进程?进程就是正在运行的程序, 一个任务就是一个进程, 进程的主要工作是管理资源, 而不是实现功能什么是线程?线程的主要
- 业务场景：前后端分离需要对接数据接口。接口测试是在postman做的，今天才开始和前端对接，由于这是我第一次做后端接口开发（第一次嘛，问题比
