更新tensorflow-gpu需要重新安裝對應cuda版本嗎?
所用的tensorflow版本太低,想升級到最新版本。是需要重新安裝對應的cuda版本,和正常安裝過程一樣?還是輸入更新命令行就可以?
主要看更新的tf-GPU版本與cuda和cudnn版本是否匹配,一般情況下要謹慎更新。目前我通過Python虛擬環境同時安裝了tf1.14和tf2,共用cuda10.0和cudnn7.4庫,以滿足不同的需求,解決歷史問題。我覺得很長時間我都不會去更新了,解決依賴問題會很頭疼。
tensorflow各個版本的CUDA以及cuDNN版本對應關係,供參考。
Linux下