linux安装tensorflow-gpu心得


2019年5月23日22点32分,记录我在ubuntu下安装tensorflowgpu的一个心得,本人菜鸡,踏坑无数,就这个坑把我折磨的电脑系统换了2遍。

关于linux、cuda、cudnn、tensorflow-gpu的安装网上一大堆,照着来没问题

我的笔记本GTX960m, Ubuntu16.04,cuda9.0 cudnn 7.0.5 tensorflow-gpu=1.5 安装成功。

但是,但是!!!重点来了!

我自己调试代码的时候发现一个问题,笔记本重启之后调通的tensorflow代码不能运行,在sess = tf.Session()的时候显示无法创建对话,错误大概是cuda 什么什么错误我也看不明白 ,就是这ZZ玩意“”“ CUDA runtime implicit initialization on GPU:0 failed. ”网上关于这个错误好像很少,我自己猜是一个bug,偶然间发现,在你第二次打开笔记本的时候,你只要先运行一个使用cuda的小代码 运行通过后,它就不报“”

tensorflow.python.framework.errors_impl.InternalError: Failed to create session.

这个错误了 ,我没找到解决办法,但是我这个先运行一个cuda的小代码确实可以在我笔记本解决这个问题,我用的代码很简单,是Warpctc的测试文件,

这个问题我也不知道解决没解决,解决了希望有人告诉我具体详细的解决办法,我是个菜鸡,不懂,那就可以试着我上面的方法,亲测真的有用!

 


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM