首页 > 其他 > 详细

实习Learning记录(二)—— 关于Tensorflow突然不能使用GPU训练的问题

时间:2021-04-13 23:38:19      阅读:24      评论:0      收藏:0      [点我收藏+]

关于Tensorflow突然不能使用GPU训练的问题

问题:

最近使用tensorflow-gpu进行训练,然后尝试使用tf_serving进行部署,然后再训练的时候发现不能使用gpu训练了,同时使用命令tf.test.is_gpu_available()检测gpu也显示False,检测不到gpu。然后就开始一步步检查,CUDAcudnntensorflow-gpu之间的适配,发现都是OK的。然后就尬住了,终于今天由组内大佬解决了这个问题。

原因:

出现上述问题的原因就是tensorflow-gpu不能和tf_serving在同一个环境下,否则就不能再使用gpu进行训练。

解决方法:

就直接把tensorflow相关的都卸载包括tf_serving,然后重新安装tensorflow-gpu就可以了。要是想部署就再新建一个环境安装tf_serving就好了,意思就是一个环境训练,一个环境进行部署。

感想:

大佬牛逼!!!!!!

实习Learning记录(二)—— 关于Tensorflow突然不能使用GPU训练的问题

原文:https://www.cnblogs.com/lzyrookie/p/14655166.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!