天天看點

cuda cudnn tensorflow對應_TensorFlow1.2~2.1各個GPU版本CUDA和cuDNN對應版本整理

最近發現很多QQ群和微信群裡經常會有人問這麼一個問題——“我安裝TensorFlow GPU版本怎麼總是提示CUDA版本不對或者cuDNN版本不對呢?”為了解決大家這個問題,我特意把TensorFlow1.2到最新版本所需要的CUDA和cuDNN對應的版本做了個整理,希望能夠對大家有幫助。

要搭建TensorFlow的GPU版本,首先需要的必備條件就是一塊能夠支援CUDA的NVIDIA顯示卡,因為在搭建TensorFlow的GPU版本時,首先需要做的一件事就是安裝其基礎支援平台CUDA和其機器學習庫cuDNN,然後在此基礎上搭建TensorFlow GPU版本。

其次還要了解一下不同的TensorFlow版本所需要對應安裝的CUDA和cuDNN版本是多少,因為在TensorFlow的GPU版本安裝過程中,如果對應的CUDA版本和cuDNN版本不正确的話,是無法正常使用GPU來進行模型訓練的。下表整理出了TensorFlow從1.2到最新版本的CUDA和cuDNN所對應的版本集合,希望對大家有幫助。

cuda cudnn tensorflow對應_TensorFlow1.2~2.1各個GPU版本CUDA和cuDNN對應版本整理

繼續閱讀