基于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
投稿猜你喜欢
- 目录前言场景分析总结前言前几天，有位朋友微信联系我，告知一个生产数据库，在机器宕机恢复后，实例启动失败，而且该实例没有做任何的高可用、容灾、
- 遇见了表中存在重复的记录的问题，直接写sql删除时最快的，才不要慢慢的复制到excel表中慢慢的人工找呢如下sql，找出重复的记录，和重复记
- 1、需求用户输入当前目录下任意文件名，程序完成对该文件的备份功能。备份文件名为xx[备份]后缀，例如：test[备份].txt。2、步骤接收
- Oracle RAC提供两种方式实现负载均衡，第一种是纯技术手段，即在用户连接时，根据系统当前的负载情况决定由哪个节点处理用户请求；第二种是
mysql分组统计并求出百分比1、mysql 分组统计并列出百分比SELECT point_id,&nb- 一、类和对象Python属于动态类型的语言，而动态语言和静态语言最大的不同，就是函数和类的定义，不是编译时创建的，而是运行时动态创建的，比方
- 第一种方法：在php.ini文件里改变display_errors和error_reporting的值，没有的直接加上; 第一处修改; di
- 一、掩膜（mask）在有些图像处理的函数中有的参数里面会有mask参数，即此函数支持掩膜操作，首先何为掩膜以及有什么用，如下：数字图像处理中
- 以select为例，如果select写在循环里，触发change事件时可能不只需要传递被选中项的值，还要传递index过去，来改变同一循环中
- 今天从网上学习了有关SQL注入的基本技能。SQL注入的重点就是构造SQL语句，只有灵活的运用SQL 语句才能构造出牛比的注入字符串。学完之后
- 本文实例讲述了mysql重复索引与冗余索引。分享给大家供大家参考，具体如下：重复索引：表示一个列或者顺序相同的几个列上建立的多个索引。冗余索
- 基础这个模块是socket的异步实现，让我们先来熟悉一下模块中的一些类和方法：1.asyncore.loop输入一个轮询循环直到通过计数或打
- 1. Shared and Exclusive Locksshared lock （译：共享锁）exclusive lock （
- 本文使用的是163邮件进行测试。注：163邮箱现在需要使用 客户端授权码 进行测试，不再支持邮箱密码进行测试。
- 目前使用MySQL的网站，多半同时使用Memcache作为键值缓存。虽然这样的架构极其流行，有众多成功的案例，但过于依赖Memcache，无
- php去掉数组的第一个值的两种删除方法：1、使用array_shift()函数删除数组的第一个值，语法“array_sh
- Update dede_addonsoft SET dxylink=REPLACE(dxylink, '.zip',
- 一、基础表单 <form > <div class="form-group"> <labe
- 在python中使用字典，格式如下：dict={ key1:value1 , key2;value2 ...}在实际访问字典值时的使用格式如
- 前言pytest配置文件可以改变pytest的运行方式，它是一个固定的文件pytest.ini文件，读取配置信息，按指定的方式去运行非tes
