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

关于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就好了,意思就是一个环境训练,一个环境进行部署。

感想:

大佬牛逼!!!!!!

posted @ 2021-04-13 20:21  Originhhh  阅读(725)  评论(0)    收藏  举报