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

问题描述

在用pip安装Tensorflow时,或者安装完import tensorflow as tf调用时,提示libcublas.so.9.0: cannot open shared object file: No such file or directory的错误。

出现原因

  • Cuda-9.0未安装或安装不正确
  • 未添加环境变量或添加完未更新
  • Tensorflow版本不支持当前Cuda或Cudnn版本

解决办法

  • 终端输入nvcc -V查看系统是否已安装Cuda-9.0,未安装则根据教程安装。

  • 检查环境变量是否添加:

终端输入:

sudo gedit .bashrc

在末尾加入:

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

一定要更新环境变量

source .bashrc

按上述操作后,如果仍未解决问题,则继续下一步。

  • 检查/usr/local/cuda-9.0/lib64路径下是否有libcublas.so.9.0文件。

如果有,在终端输入:

sudo ldconfig /usr/local/cuda-9.0/lib64

上述三部之后仍然提示出错,则需要考虑安装的Tensorflow版本是否支持当前安装的Cuda和Cudnn版本,如果不支持,则需要回滚至旧的版本。

sudo pip --upgrade tensorflow-gpu==型号
    原文作者:Edwin_dl
    原文地址: https://www.jianshu.com/p/6c649b1969c7
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞