1. 程式人生 > 程式設計 >匯入tensorflow:ImportError: libcublas.so.9.0 報錯

匯入tensorflow:ImportError: libcublas.so.9.0 報錯

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

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

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

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

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

安裝完正確的版本後,確認你在你的~/.bashrc(或者~/.zshrc)檔案中加入了下面環境變數

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

如圖所示:

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

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

pip install --upgrade tensorflow-gpu==1.4

以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。