之前在pytorch上写的一个模型，昨天在tf上写了一遍，但是gpu利用率特别低，pytorch在windows和ubuntu都OK，但是tf在windows上完全不行，在tensorboard里面查看了session run每个节点的时间，前向还比较快，后向就很慢，特别是靠近输入的那几个convolution layer。
然后写了个很浅的net，也是很慢，这时候基本是猜到了是操作系统的问题，换到ubuntu测试了下TF，OK了。
当然ubuntu测试也遇到恶心到问题。ubuntu18只能装cuda9，最开始想那就装个最新的cuda9.2吧，不行，import tf的时候提示它想找cuda9.0的lib。然后我开始有点作死了，仍然装9.1（因为同事装上了9.1），不行之后把9.1，9.2都删除了，然后重新装9.0+cudnn7 可以用tf了……之前同事能装上可能是它用的conda吧，所以别人的经验不能照搬，还是要多思考，多看报错信息的来源
至于之前的windows tf为什么不行，可能是tf版本太低？或者cpu太好与低版本的cuda8不match？
感觉这一天真的没做啥事啊！真的有点儿效率低啊！
网友评论