使用Tensorflow-GPU禁用GPU设置(CPU与GPU速度对比)
作者:ArchonGum 时间:2022-06-10 19:10:11
禁用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则记录一些日志。
来源:https://blog.csdn.net/c13232906050/article/details/83505563
标签:Tensorflow,GPU,CPU,速度


猜你喜欢
c++基础语法:虚继承
2024-01-23 10:01:00
php 生成随机验证码图片代码
2024-05-02 17:33:19
Python趣味实例,实现一个简单的抽奖刮刮卡
2023-07-20 18:59:41

超实用的全新JavaScript事件Scrollend实例详解
2024-04-16 10:27:49

详解Django之admin组件的使用和源码剖析
2022-10-07 06:36:27

Python中tell()方法的使用详解
2021-06-29 16:21:59
使用Python+wxpy 找出微信里把你删除的好友实例
2023-05-09 05:12:28

Python学习之基础语法介绍
2022-03-22 22:08:54

Python定时任务随机时间执行的实现方法
2023-11-23 18:40:21
网站中美好的细节
2011-07-13 18:43:07
SpringBoot Logback日志记录到数据库的实现方法
2024-01-16 11:58:11

Python实现敏感词过滤的4种方法
2021-10-01 06:21:08

一文详解Python定时任务触发
2021-05-13 14:27:02

Go中如何使用set的方法示例
2024-04-28 10:47:19
使用Python实现一个蔡徐坤大战篮球的小游戏(推荐)
2022-03-07 19:02:13

Golang中实现数据脱敏处理的go-mask包分享
2024-02-12 17:24:55
php 进阶:实现无限分类第1/4页
2023-11-17 21:04:55
python实现简单tftp(基于udp协议)
2021-01-30 15:01:23

在Django中创建URLconf相关的通用视图的方法
2021-09-17 06:07:58
设计的技术含量
2009-01-12 18:20:00