tensorflow can't open CUDA library libcupti.so.8.0 or libcupti.so.9.0

在tensorflow 1.12 运行mnist example 时前90个step正常,但是到step 90的时候报错,无法打开usr/local/cuda/lib64/libcupti.so.8.0 的问题,笔者猜测可能是新旧版本默认寻找libcuti.so.8.0的路径变了,或者存放相应文件的目录做了改变,通过查看cuda安装的文件夹可以看到该文件所在的目录并没有需要的目标文件。

解决办法:

在环境变量中添加libcupti.so.8.0的路径,比如可以使用如下语句解决该问题:

sudo gedit /etc/enviroment

添加以下内容:

LD_LIBRARY_PATH="/usr/local/cuda-8.0/lib64:/usr/local/cuda-8.0/extras/CUPTI/lib64"  (对应安装的位置)

 

重启,再次运行

Accuracy at step 90: 0.8911
I tensorflow/stream_executor/dso_loader.cc:135] successfully opened CUDA library libcupti.so.8.0 locally
Adding run metadata for 99

 

发现此时可以正确打开需要的文件,运行到刚才的step 90顺利通过,继续运行。

 

------------------------------------------------------------------  更新 ------------------------------------------------------------

重装系统ubuntu 16.04后,在pycharm中,无法正常导入和使用tensorflow,但是在命令行却可以正常导入。以前用上面方法是可以解决问题的,但是新系统下竟然不行了,还不太明白原因,但庆幸的是,问题依然是可以解决的。此时需要用下面命令即可彻底解决问题,不需要在pycharm中配置环境变量了,

  1. sudo gedit /etc/ld.so.conf.d/cuda.conf (add)
    1. /usr/local/cuda/lib64
    2. /lib
  2. sudo ldconfig -v
  3.  

另一种解决方法:

或者必须在命令行中打开pycharm(在这之前需要先建立命令行全局的链接,比如sudo ln -s /home/jia/Software/pycharm-2018.3.4/bin/pycharm.sh /usr/local/bin/pycharm)。

  • 4
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值