1. 程式人生 > >安裝tensorflow-gpu過程中和nvidia driver 匹配的問題

安裝tensorflow-gpu過程中和nvidia driver 匹配的問題

【本地環境】

作業系統:Window server 2012 R2 64bits

NVIDIA版本:Tesla K80,driver版本用nvidia-smi命令檢視是376.51

這樣的情況下自己安裝tensorflow-gpu時太大意了,直接conda install tensorflow-gpu,結果自動安裝了 9.0 版本的cudatoolkit 和7.1.4版本的cudnn,然後程式就報了錯誤(忘記截圖了):CUDA driver version is insufficient for CUDA runtime version

原因就是cudatoolkit 9.0 和 nvidia driver 376.51 不匹配

把cuda和tensorflow-gpu降級成下屬版本即可:

Cudatoolkit版本:8.0

cudnn版本:6.0

Tensorflow-gpu版本:1.1.0

當然也可以升級nvidia driver 至和 cudatoolkit匹配的版本

具體nvidia driver匹配哪個版本 cudatoolkit 可以在nvidia官網查到download diver

參考連結如下: