linux安裝keras+tensorflow-gpu步驟


最近剛剛開始接觸深度學習,感覺需要用一下博客記錄一下平時遇見的坑和解決方案。

最近從網上下載了一個代碼是keras+tensorflow的,第一次運行python代碼有點激動,中間遇見了一些坑,記錄一下解決方案。

最主要的是keras和tensorflow-gpu的版本不匹配造成的。比如會有錯誤:'eras.backend' has no attribute 'contro_flow_ops'

1.創建虛擬環境,防止很多框架放在一個主目錄下在后面操作中太混亂:括號是解釋,運行命令的時候刪除。

conda create -n Ma(虛擬環境名稱)python==3.6.7(這個環境以前以為要和以前安裝的python版本對應,其實是不必要的,這個版本可以根據代碼要求設定,比如可以3.5或3.6.)

2.安裝tensorflow,因為自己用的服務器可以使用GPU,所以這里安裝tensorflow-gpu版本:

conda install tensorflow-gpu==1.12.0

(這一步會自動安裝 cudatoolkit 9.2 和 cudnn 7.6.0)

3.安裝keras

conda install keras==2.2.4

4.降低一下numpy的版本

conda numpy==1.16.0


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM