InternalError: cudaGetDevice() failed. Status: CUDA driver version is insufficient for CUDA runtime version


安裝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,就不會出現上面的錯誤了.


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM