tensorflow2.0 设置GUP

用tensorflow2.0设置

import tensorflow as tf#查看GPU列表
gpus = tf.config.experimental.list_physical_devices(device_type='GPU')#查看CPU列表
cpus = tf.config.experimental.list_physical_devices(device_type='CPU')#设置可见设备 即 选择GPU CPU 范围
tf.config.experimental.set_visible_devices(devices=gpus[0:2],device_tyoe='GPU')#设置需要动态申请时
for gpu in gpus:tf.config.experimental.set_memory_growth(device=gpu,True)

设置环境变量来设置

import osos.environ['CUDA_VISIBLE_DEVICES']="2,3"

tensorflow2.0 设置GUP相关推荐

【深度学习】(6) tensorflow2.0使用keras高层API
各位同学好,今天和大家分享一下TensorFlow2.0深度学习中借助keras的接口减少神经网络代码量.主要内容有: 1. metrics指标:2. compile 模型配置:3. fit 模型训练 ...
【TensorFlow2.0】(7) 张量排序、填充、复制、限幅、坐标选择
各位同学好,今天和大家分享一下TensorFlow2.0中的一些操作.内容有: (1)排序 tf.sort().tf.argsort().top_k():(2)填充 tf.pad():(3)复制 tf ...
【TensorFlow2.0】(2) 创建tensor的方法
各位同学好,今天和大家分享一下TensorFlow2.0中的tensor变量的创建方法.内容有: (1) 通过numpy和list创建tensor:(2) 创建全部为某个值的tensor:(3) 随机 ...
Tensorflow2.0安装（win10系统cpu版本）
前提条件:已经安装好Anaconda,本文重点不详细介绍,安装Anaconda见这篇文章: [Anaconda教程01]怎么安装Anaconda3 - 知乎默认安装完Anaconda 步骤一:启动A ...
【TensorFlow2.0】数据读取与使用方式
大家好,这是专栏<TensorFlow2.0>的第三篇文章,讲述如何使用TensorFlow2.0读取和使用自己的数据集. 如果您正在学习计算机视觉,无论你通过书籍还是视频学习,大部分的教 ...
【小白学PyTorch】扩展之Tensorflow2.0 | 21 Keras的API详解（下）池化、Normalization
<<小白学PyTorch>> 扩展之Tensorflow2.0 | 21 Keras的API详解(上)卷积.激活.初始化.正则扩展之Tensorflow2.0 | 20 TF ...
【小白学PyTorch】扩展之Tensorflow2.0 | 21 Keras的API详解（上）卷积、激活、初始化、正则...
[机器学习炼丹术]的学习笔记分享 <<小白学PyTorch>> 扩展之Tensorflow2.0 | 20 TF2的eager模式与求导扩展之Tensorflow2.0 | ...
【小白学PyTorch】扩展之Tensorflow2.0 | 20 TF2的eager模式与求导
[机器学习炼丹术]的学习笔记分享 <<小白学PyTorch>> 扩展之Tensorflow2.0 | 19 TF2模型的存储与载入扩展之Tensorflow2.0 | 18 ...
Tensorflow2.0(Keras)转换TFlite
Tensorflow 2.0(Keras)转换TFlite 目录 Tensorflow 2.0(Keras)转换TFlite 1. TensorFlow Lite 指南 (1)TensorFlow L ...

tensorflow2.0 设置GUP

tensorflow2.0 设置GUP相关推荐

最新文章

热门文章