基于Tensorflow使用CPU而不用GPU问题的解决
作者：JR_Chan 发布时间：2022-01-01 22:53:08
标签：Tensorflow,CPU,GPU
之前的文章讲过用Tensorflow的object detection api训练MobileNetV2-SSDLite，然后发现训练的时候没有利用到GPU，反而CPU占用率贼高（可能会有Could not dlopen library 'libcudart.so.10.0'之类的警告）。经调查应该是Tensorflow的GPU版本跟服务器所用的cuda及cudnn版本不匹配引起的。知道问题所在之后就好办了。
检查cuda和cudnn版本
首先查看cuda版本：
cat /usr/local/cuda/version.txt
以及cudnn版本：
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
重新安装对应版本Tensorflow
根据前面查看得到的cuda和cudnn版本，到Tensorflow官网查看对应的Tensorflow-GPU版本，然后用conda install tensorflow-gpu=[version]重新安装（把[version]换成对应的版本比如1.12）就OK了。
来源：https://blog.csdn.net/JR_Chan/article/details/101374175
0
投稿猜你喜欢
wechat_sender 是基于 wxpy 和 tornado 实现的一个可以将你的网站、爬虫、脚本等其他应用中各种消息 （日志、报警、运- 今天弄了一天，总算把win2003下的问题给解决了， LoadModule php5_module E:\server\php528\php
- 首先,这片文章纯粹是我的个人经验之谈,适用于我常见的环境及项目中。个人建议,数据库字符集尽量使用utf8(HTML页面对应的是utf-8),
- 其实golang用一个函数可以构建一个并发队列，现在编写一个灵活可控的队列程序先定义一个工作type Worker struct { &nb
- 写这个的目地，主要是系统理下目前产品设计的流程,提醒自己尽量去避免一些常见的问题，也能让大家系统的了解天极网的产品设计流程。当然，不保证任何
- 问题：在论坛发表回复时出现“The table is full”的提示，字面意义上是数据表已满的意思。因为很少有开发者遭遇单一表超过4G的情
- YSlow是yahoo美国开发的一个页面评分插件，非常的棒，从中我们可以看出我们页面上的很多不足，并且可以知道我们改怎么却改进和优化。仔细研
- 今天，在家试试django的model的设置，如何设置的联合主键，我经过查资料和实践，把结果记录如下：例如：class user(Model
- 我们前面介绍的Selenium的WebDriver测试方法是基于网页来进行的。之前的例子都是用简单的网页来操作，可能体会不到网页的加载过程，
- 浏览器：IE ，不支持firefoxfilter视觉滤镜的种类:Alpha(透明度) Blur(模糊) Chroma(指定颜色透明) Dro
- 1、需求当工作在UNIX Shell下时，我们想使用常见的通配符模式（即：.py，Dat[0-9].csv等）来对文本做匹配。2、解决方案f
- 周末出去爬山，照了一大堆照片回来，照片同时存储为jpg和DNG格式，我用adobe bridge将dng格式的照片中要保留的筛选出来后，就不
- 本文实例讲述了Python常用字符串替换函数strip、replace及sub用法。分享给大家供大家参考，具体如下：今天在做一道今年秋季招聘
- 1 前言在前面的章节中我们牛刀小试，一直在使用python爬虫去抓取数据，然后把数据信息存放在数据库中，至此已经完成了基本的基本信息的处理，
- 本文根据自己初学经验编写的使用xlwt模块设置单元格的一些基本样式，如设置单元格的背景颜色，下框线，字体，字体的颜色，设置列宽行高，插入简单
- ---- ORACLE数据库由数据文件，控制文件和联机日志文件三种文件组成。由于磁盘空间的变化，或者基于数据库磁盘I/O性能的调
- Dataframe使用loc取某几行几列的数据：print(df.loc[0:4,['item_price_level',&
- 本文为大家分享了python链表的基础概念和基础用法，供大家参考，具体内容如下一、什么是链表链表是由多个不同的节点组成，每个节点通过指针区域
- php中主要用到的就是要用到fread()和fwirte()。而静态页面生成了之后，就会牵扯到修改的问题。这里可以用到正则匹配的方法来替换模
- 本文实例讲述了PHP实现的获取文件mimes类型工具类。分享给大家供大家参考，具体如下：<?php/* * Copyright 201
