所用的tensorflow版本太低,想升級到最新版本。是需要重新安裝對應的cuda版本,和正常安裝過程一樣?還是輸入更新命令行就可以?


主要看更新的tf-GPU版本與cuda和cudnn版本是否匹配,一般情況下要謹慎更新。目前我通過Python虛擬環境同時安裝了tf1.14和tf2,共用cuda10.0和cudnn7.4庫,以滿足不同的需求,解決歷史問題。我覺得很長時間我都不會去更新了,解決依賴問題會很頭疼。

tensorflow各個版本的CUDA以及cuDNN版本對應關係,供參考。

Linux下

Windows下

參考連接:

https://www.tensorflow.org/install/source#common_installation_problems?

www.tensorflow.org

https://www.tensorflow.org/install/source_windows?

www.tensorflow.org


如果空間足夠的話,可以用conda或者docker之類的環境

一份代碼一份環境,不用考慮那麼多


cuda和cudnn現在也可以用指令安裝,不需要以前手動安裝了,很方便,更新是需要安裝對應cuda的


不要更新,最好是一一對應


實際上只要夠用沒必要更新TensorFlowGPU版本,推薦1.14


推薦閱讀:
相關文章