Win10下Tensorflow-gpu使用cpu训练而不是gpu加速
问题1:明明安装好了tensorflow-gpu,但是运行代码时无法使用gpu加速
明明安装好了tensorflow-gpu,但是运行代码时无法使用gpu加速,运行tensorflow代码时cpu占满还是跑的很慢,同时tf.test.is_gpu_available()=false。在网上看了很多教程,有人说是tensorflow与tensorflow-gpu的版本不同问题,要求tensorflow-gpu版本比tensorflow版本高,同时要求电脑在装tensorflow之后装tensorflow-gpu。
因此我就尝试了很多次,发现都不成功,tf.test.is_gpu_available()=false一直都是这样。因为我一直用pytorch跑代码,pytorch可以用gpu加速但是tensorflow不可以,说明tensorflow配置可能有问题。
原因分析:
Tensorflow对CUDA和CuDNN的要求很严格,如果gpu用不了建议首先根据这个博主的博客对照一下tensorflow版本对应问题。
附上我觉得总结很好的博主的博客:https://blog.csdn.net/weixin_44560088/article/details/117457619?spm=1001.2014.3001.5502
例如:tensorflow_gpu-2.3.0必须要求python为3.5-3.8;CuDNN=7.6(7版本开头);CUDA=10.1。
解决方案:
1.首先查看自己CUDA的版本
在Anaconda中输入nvcc -V查看自己的cuda版本,比如我的就是10.1版本,这样看最靠谱。
我之前是CUDA10.2版本,没有对应的tensorflow,因此尝试了很多次都不成功,所以我重新找了一个CUDA版本和CUDNN版本进行下载,在没有删除10.2的情况下又安装了一个10.1版本。
2.查看版本对应表选择一个自己的CUDA版本下的Tensorflow版本
3.在Anaconda Navigator中把原来的Tensorflow虚拟环境删除掉(右下角remove)
4.下载好CUDA10.1和CuDNN7.6.5的压缩包(这里一定要下CuDNN版本7开头的压缩包而不是8开头的,见问题2)
将CUDA10.1与原来的10.2版本的环境分开,比如10.2在C盘,为了不出什么问题,我把10.1安装在了D盘,首先在D盘组织好这样的架构D:\CUDA\NVIDIA corporation\CUDA Samples\v10.1和D:\CUDA\NVIDIA GPU Computing Toolkit\CUDA\v10.1,根据CUDA的默认C盘安装路径将安装路径修改到对应的D盘,方便后续查找文件。
将CuDNN解压缩后将文件复制到D:\CUDA\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin路径下替换掉对应的文件。
5.在Anaconda中创建对应的tensorflow虚拟环境(我给他起名字叫tf)
conda create -n tf python=3.8
创建完毕后,使用pip安装tensorflow-gpu
pip install tensorflow-gpu==2.3.0
等待其安装成功后,就可以依次在Anaconda prompt中输入以下代码
activate tf
python
import tensorflow as tf
print(tf.test.is_gpu_available())
如果看到True就表示成功了!
问题2:Could not load dynamic library ‘cudnn64_7.dll’
原因分析:
没有关注CuDNN的版本,我在第一次安装时直接下载了CuDNN 8开头的版本,导致在bin文件夹中找不到cudnn64_7.dll只能找到cudnn64_8.dll,这就是没有认真读题的原因!!!
解决方法:下载CuDNN 7开头的版本解压去bin文件粘贴替换即可
当你完成所有这些之后,重启你的编译器/终端,输入上述测试tensorflow-gpu能否使用的命令,就可以惊喜的发现!它成功了!接下来就可以用gpu快乐的跑代码了~
此处特别鸣谢我的师弟——阿刚!在他的帮助下我们一起完美解决了这个问题~
Win10下Tensorflow-gpu使用cpu训练而不是gpu加速相关推荐
- GPU和CPU是什么?GPU和CPU什么关系?GPU会取代CPU吗?
GPU和CPU是什么?GPU和CPU什么关系?GPU会取代CPU吗? GPU是什么?与显卡是什么关系?安装在什么地方?有单独的GPU板卡吗? CPU和GPU区别是什么?GPU会取代CPU吗? GPU能 ...
- Win10下Tensorflow(GPU版)安装趟坑实录,成功了。
这篇文章不指导你具体怎么安装,其他博客大把安装方法,我只告诉你,具体的版本不要看其他人的,每个人用的CUDA可能都不同,越看越乱,直接看运行后系统的提示,我整了一天没有搞定,最后还是按照系统提示搞定的 ...
- win10下tensorflow库gpu运行环境配置实操
TensorFlow NVIDIA显卡驱动 CUDA套件 cuDNN驱动 安装 测试 近日,在github上闲逛时发现了一个非常有趣的一个实验,地址如下DeepLearningFlappyBird,见 ...
- WIN10下TensorFlow GPU版安装顺序
1.更新驱动 2.先安装Anaconda,创建一个虚拟环境 4.再安装TensorFlow-GPU 5.到虚拟环境所在的目录里寻找build_info.py文件,一般如果全部默认安装的话在: C:\U ...
- TensorFlow2.0选择GPU或CPU训练
注意:以下所有程序都要在导入TensorFlow和keras包之前加入才可以 选择CPU: import os os.environ["CUDA_DEVICE_ORDER"] = ...
- kera tensorflow只使用cpu训练
原因是在gpu下超出显存,不得已在cpu下训练模型 该链接中有,点击打开链接 import os os.environ["CUDA_DEVICE_ORDER"] = "P ...
- WIN10 下 Tensorflow 环境搭建
TensorFlow开发环境搭建 1.安装python 2.安装eclipse+pydev插件 3.安装Tensorflow等python插件 这里介绍基于eclipse的Tensorflow环境搭建 ...
- 在Win10下删除快捷方式小图标,而不伤及其他注册表项的方法
将以下代码保存为.reg,添加到注册表,注销之后就生效了 Windows Registry Editor Version 5.00 [HKEY_CLASSES_ROOT\lnkfile] " ...
- linux下cpu opencl加速,GPU挑战CPU!详解CUDA+OpenCL威力
众所周知,GPU拥有数十倍于CPU的浮点运算能力,但如此强大的实力多数情况下只能用来玩游戏,岂不可惜?因此近年来业界都在致力于发掘GPU的潜能,让它能够在非3D.非图形领域大展拳脚. 1999年,首颗 ...
最新文章
- Python 技术篇-requests发送post请求传文件给flask服务端报405、400问题解决方法
- android更改menu字体颜色,Android动态修改menu颜色
- HTTP错误404.3-Not Found
- CodeForces - 1409F Subsequences of Length Two(dp)
- vue 带全选和多选的表格怎么写_EXCEL五分钟,批量制作带照片的工地出入证
- 总裁徐雷“接管”京东 但拍板的仍是刘强东
- linux命令详解——tar
- Deploy a replica set
- Android强大log框架Logger使用详解
- python编程读取文件内容_python编程从入门到实践:读取整个文件和创建包含文件各行的内容...
- windows10批量解压缩文件-用WinRAR
- 青少年c语言培训,青少年信息学奥赛培优教程·入门篇(2020年01月)
- 12星座的出生年月日性格_十二星座出生日期
- 揭秘无聊程序猿的趣味人生|斑鸠职业
- 织梦dede列表分页样式
- xshell 运行python脚本
- 响应式织梦模板家政服务保姆月嫂类网站
- 怎么实现角色权限的分配_Choerodon猪齿鱼实践之角色管理
- python画图柱状图修改相邻两个柱之间的间隔
- 如何开展网安XX行动防护工作