tensorflow中gpu和cpu切换

两者同时存在时使用gpu
要想使用cpu，在代码里添加如下命令：

matrix1=tf.constant([[3.,3.]])
matrix2=tf.constant([[2.],[2.]])
with tf.Session() as ses:with tf.device("/cpu:0"):product=tf.matmul(matrix1,matrix2)

说明一下：
“/cpu:0” The CPU of your machine
“/gpu:0” The GPU of yout machine ,if you have one
如果要换成gpu的话，照着改可以了
或
os.environ[“CUDA_VISIBLE_DEVICES”] = "-1"
一行即可

tensorflow中gpu和cpu切换相关推荐

深度学习PyTorch，TensorFlow中GPU利用率较低，CPU利用率很低，且模型训练速度很慢的问题总结与分析
在深度学习模型训练过程中,在服务器端或者本地pc端,输入nvidia-smi来观察显卡的GPU内存占用率(Memory-Usage),显卡的GPU利用率(GPU-util),然后采用top来查看CPU ...
深度学习PyTorch，TensorFlow中GPU利用率较低,使用率周期性变化的问题
在用tensorflow训练神经网络时,发现训练迭代的速度时而快时而慢,监督的GPU使用率也是周期性变化,通过了解,发现原因是: GPU在等待CPU读取,预处理,并传输数据过来,因此要提高GPU的使用 ...
深度学习PyTorch、TensorFlow中GPU利用率与内存占用率很低的问题
上周,在一个使用Pytorch搭建的目标训练项目中,训练时,通过使用命令行执行NVIDIA-SMI(仅支持英伟达显卡)命令发现GPU的利用率基本一直停留在0%,并且显存占用率也较低.CSDN上有一篇分 ...
tensorflow中GPU的设置
设置使用哪块GPU: import os os.environ['CUDA_VISIBLE_DEVICES'] = '0' # 使用 GPU 0 os.environ['CUDA_VISIBLE_DE ...
查看安装的tensorflow是哪个版本，查看Tensorflow是GPU还是CPU
>>>import tensorflow as tf >>>tf.__version__ '2.1.0' >>> tf.__path__ ['C: ...
opengl中gpu与cpu交互_OpenGL 环境配置与教程推荐
我们先了解一下 OpenGL 是什么: 摘自 Wikipedia: Open Graphics Library (OpenGL) is a cross-language, cross-platform ...
tensorflow中GPU相关设置解决显存不足
1.显存不足时报错如下: E tensorflow/stream_executor/cuda/cuda_dnn.cc:359] could not create cudnn handle: CUDNN ...
TensorFlow GPU与CPU运行速度比较
测试TensorFlow采用GPU或CPU对于同一任务的运行速度测试. 第一次使用TensorFlow,第2个月使用python,不是CS行业. 测试环境软件环境:windows10,VScode, ...
Tensorflow + PyTorch 安装（CPU + GPU 版本）
目录一.Anaconda 安装二.安装 TensorFlow-CPU 1.配置环境 2.安装 Tensorflow 三.安装TensorFlow-GPU 1.是否可安装GPU版Tensorflow ...
tf.device()指定tensorflow运行的GPU或CPU设备
在tensorflow中,我们可以使用 tf.device() 指定模型运行的具体设备,可以指定运行在GPU还是CUP上,以及哪块GPU上. 设置使用GPU 使用 tf.device('/gpu:1' ...

tensorflow中gpu和cpu切换

tensorflow中gpu和cpu切换相关推荐

最新文章

热门文章