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