当前位置: 首页 > 编程笔记 >

使用Tensorflow-GPU禁用GPU设置(CPU与GPU速度对比)

公羊涛
2023-03-14
本文向大家介绍使用Tensorflow-GPU禁用GPU设置(CPU与GPU速度对比),包括了使用Tensorflow-GPU禁用GPU设置(CPU与GPU速度对比)的使用技巧和注意事项,需要的朋友参考一下

禁用GPU设置

# 在import tensorflow之前
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '-1'

CPU与GPU对比

显卡:GTX 1066

CPU

GPU

简单测试:GPU比CPU快5秒

补充知识:tensorflow使用CPU可以跑(运行),但是使用GPU却不能用的情况

在跑的时候可以让加些选项:

with tf.Session(config=tf.ConfigProto(allow_soft_placement=True, log_device_placement=True))

其中allow_soft_placement能让tensorflow遇到无法用GPU跑的数据时,自动切换成CPU进行。

log_device_placement则记录一些日志。

以上这篇使用Tensorflow-GPU禁用GPU设置(CPU与GPU速度对比)就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持小牛知识库。

 类似资料:
  • 我正在尝试创建和训练一个CNN模型。但每次我运行代码时,tensorflow并没有使用GPU,而是使用CPU。我已经安装了tensorflow的最新版本。附上以下详细信息。 在运行时,我得到以下带有警告消息的输出。(平台:VS代码) 2021-07-28 15:35:13.163991: W tenstorflow/core/common_runtime/bfc_allocator.cc:337]

  • 我目前正在尝试训练一个聊天机器人,更具体地说是这个。但是,当我开始训练聊天机器人时,它使用了我100%的CPU和大约10%的GPU。有人知道为什么吗。 我已经安装了,并确保我有正确的CUDA和cuDNN版本。我还确保我没有安装基本的pip包。我的GPU也有最新的英伟达驱动程序。我也尝试过卸载和重新安装我所有的驱动程序,CUDA,cuDNN,tensorflow gpu和它所有的依赖和python本

  • 我正在尝试测试Aparapi的性能。我看到过一些博客,其中的结果显示,Aparapi确实在做数据并行操作的同时提高了性能。 但我在测试中没有看到这一点。这里是我所做的,我写了两个程序,一个使用Aparapi,另一个使用普通循环。 方案1:在Aparapi 程序2:使用循环 程序1需要大约330ms,而程序2只需要大约55ms。我是不是做错什么了?我在Aparpai程序中打印出了执行模式,它打印出的

  • 我在我的Ubuntu 19.04笔记本电脑上使用设置了TensorFlow。所有依赖项,如CUDA,CUDNN都安装并工作。但是,当导入TensorFlow并检查给我False。我尝试过完全卸载和重新安装TensorFlow,这不起作用。输出: 2019-06-27 14:06:18.359739: I tenstorflow/core/平台/cpu_feature_guard.cc:142]您的

  • 如何在Python 3.6 x64中使用TensorFlow GPU版本而不是CPU版本? Python正在使用我的CPU进行计算 我可以注意到,因为我有一个错误: 您的CPU支持该TensorFlow二进制文件未编译使用的指令:AVX2 我已经安装了tensorflow和tensorflow gpu。 如何切换到GPU版本?

  • 本文向大家介绍已安装tensorflow-gpu,但keras无法使用GPU加速的解决,包括了已安装tensorflow-gpu,但keras无法使用GPU加速的解决的使用技巧和注意事项,需要的朋友参考一下 问题 我们使用anoconda创建envs环境下的Tensorflow-gpu版的,但是当我们在Pycharm设置里的工程中安装Keras后,发现调用keras无法使用gpu进行加速,且使用的