ICode9

精准搜索请尝试: 精确搜索
首页 > 系统相关> 文章详细

Linux服务器tensorflow跑程序不调用GPU而用CPU

2021-11-13 19:04:56  阅读:243  来源: 互联网

标签:调用 Linux 程序 tensorflow Usage GPU CPU


在这里记录一下自己在服务器上跑tensorflow程序一直不调用GPU而用CPU的解决过程:
一开始跑程序,我发现特别的慢,跑了一天才跑了三分之一,后面用nvidia-smi查看GPU使用情况,发现压根没有用GPU跑,然后再用top查看,发现一直再用CPU跑。。。

首先在网上查阅资料发现自己没有在环境里装tensorflow-gpu版,再次运行程序,接着用nvidia-smi查看,这个程序的进程出现在了里面,但是。。。却没有GPU-Util却为0%,Memory-Usage也只为62MiB/11441MiB,很是疑惑。
接着检查cuda版本,为10.1,网上说和tensorflow1.15.0不匹配,好吧,我再重新在我的用户下配cuda10.0(教程)。配置完毕后,再次运行,结果还是一样,GPU-Util却为0%,Memory-Usage也只为62MiB/11441MiB。然后我又配置了cuda9.0,一样还是不行。后面再网上查阅资料说时batch_size太小有可能会导致这种情况,我又重新调batch_size重新跑,但是还是一样结果。
经过这么多次失败,但最终还是成功了!原因竟然是tensorflow-gpu要用conda来下载,而不是pip(额,,,,很是无语,完全没有想到会是这个原因),最终成功调用了GPU(太难了)。

标签:调用,Linux,程序,tensorflow,Usage,GPU,CPU
来源: https://blog.csdn.net/weixin_43254131/article/details/121308212

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有