1. 程式人生 > >[解決方案]ImportError: libcublas.so.9.0: cannot open shared object file

[解決方案]ImportError: libcublas.so.9.0: cannot open shared object file

用pip直接安裝tensorflow之後,import tensorflow的時候出現如題的錯誤。在網上找了一下,發現可能有兩種原因。

  1. tensorflow版本與CUDA版本不匹配。這種情況下,一般直接升/降級tensorflow就行。如果更換CUDA,比較麻煩,安裝之後還是不要變了。

  2. 可能是CUDA安裝之後的路徑沒有配置好。參考新增連結描述,執行以下命令即可。

    export LD_LIBRARY_PATH=LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64/
    source .bashrc