安裝TensorFlow 2.0.0以后,運行出現了下面的錯誤:
tensorflow.python.framework.errors_impl.InternalError: cudaGetDevice() failed. Status: CUDA driver version is insufficient for CUDA runtime version
這是由於CUDA驅動版本不滿足CUDA運行版本造成的,之前為了安裝TensorFlow 2.0.0,升級了CUDA運行版本到CUDA 10.0,但是CUDA驅動版本並沒有升級,從而造成了這個錯誤.CUDA驅動版本和CUDA運行版本應該滿足下面這張表的要求(https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html):
運行nvidia-smi后,發現CUDA驅動版本是385.54,不滿足上面的要求.
於是直接用Windows8.1的設備管理器,用自動搜索更新的驅動程序軟件升級CUDA驅動版本:
升級結束以后,再運行nvidia-smi,顯示CUDA驅動版本已經升級到了416.34,此時運行TensorFlow 2.0.0,就不會出現上面的錯誤了.