最近使用tensorflow-gpu
进行训练,然后尝试使用tf_serving
进行部署,然后再训练的时候发现不能使用gpu训练了,同时使用命令tf.test.is_gpu_available()
检测gpu也显示False,检测不到gpu。然后就开始一步步检查,CUDA
、cudnn
、tensorflow-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