溫馨提示×

溫馨提示×

您好,登錄后才能下訂單哦!

密碼登錄×
登錄注冊×
其他方式登錄
點擊 登錄注冊 即表示同意《億速云用戶服務條款》

導入tensorflow:ImportError: libcublas.so.9.0報錯怎么辦

發(fā)布時間:2021-07-12 13:38:27 來源:億速云 閱讀:185 作者:小新 欄目:開發(fā)技術

小編給大家分享一下導入tensorflow:ImportError: libcublas.so.9.0報錯怎么辦,相信大部分人都還不怎么了解,因此分享這篇文章給大家參考一下,希望大家閱讀完這篇文章后大有收獲,下面讓我們一起去了解一下吧!

錯誤:ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory

問題:找不到cuda9.0的版本。

出現(xiàn)該錯誤的主要原因:cuda未安裝或者cuda的版本有問題

這個錯誤在安裝tensorflow時經常會出現(xiàn),但是在官方的常見問題中沒有提到, 如果你使用下面的方法沒有解決的話,可以在評論中留言。

對于tensorflow 1.7版本,只接受cuda 9.0(9.1也不可以?。?,和cudnn 7.0,所以如果你安裝了cuda9.1和cudnn7.1或以上版本,那么你需要重新安裝9.0和7.0版本。

安裝完正確的版本后,確認你在你的~/.bashrc(或者~/.zshrc)文件中加入了下面環(huán)境變量

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64
export PATH=$PATH:/usr/local/cuda-9.0/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-9.0

如圖所示:

導入tensorflow:ImportError: libcublas.so.9.0報錯怎么辦

之后重啟電腦即可,將不會在提示ImportError: libcublas.so.9.0的錯誤。

如果你的cuda是8.0,那么你可以使用下面的指令將tensorflow的版本回滾到1.4版本,

pip install --upgrade tensorflow-gpu==1.4

以上是“導入tensorflow:ImportError: libcublas.so.9.0報錯怎么辦”這篇文章的所有內容,感謝各位的閱讀!相信大家都有了一定的了解,希望分享的內容對大家有所幫助,如果還想學習更多知識,歡迎關注億速云行業(yè)資訊頻道!

向AI問一下細節(jié)

免責聲明:本站發(fā)布的內容(圖片、視頻和文字)以原創(chuàng)、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯(lián)系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。

AI