今天要迁移项目,本来我一直在本地跑的用anaconda3.6主里面,原来创建虚拟的环境方便迁移,这下我只有重新创建新环境重下包进行迁移,但是包都下好了却出现这样奇怪的错误,百度了说是找不到GPU,可是我之前都能跑啊换个环境没理由GPU没有跑不了? 按照有些帖子在session那里写成这样:
config = tf.ConfigProto(allow_soft_placement = True)
sess = tf.Session(config = config)
又报错:Conv2DCustomBackpropInputOp only supports NHWC.
不对,一定哪里不一样了我就看原来的运行日志发现了了tensorflow下的core/common_runtime,可是神奇的是tensorflow-gpu不一样,肯定新手按安装tensorflow安装了很多次有没有卸载
我把这个和虚拟环境安装的tensorflow-gpu里面查看居然不一样,随后感觉可能是安装有问题,一般安装命令是pip install tensorflow1.10.0或者conda install tensorflow1.10.0直觉不能再是这两个了,因为都不能使运行起程序 然后又百度tensorflow-base-gpu看到安装 多亏了这篇博客 最后卸载虚拟环境的tensorflow再用这句重新安装!!!
conda create --name tensorflow-gpu python=3.6
可以看到有了common_runtime,重新运行代码也可以跑起来了搞定!!!
不正常的tensorflow-gpu 我需要的tensorflow-gpu 也就是include下面还有
|