您的位置:首页 > 其它

tensorflow运行mnist例程第99次迭代cuda出错

2017-06-27 14:22 239 查看
新手,在run 例程中的mnist_with_summaries.py的时候在第99次迭代出现问题:

找不到libcupti.so文件,以为是cuda没配置好,试了

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64"
export CUDA_HOME=/usr/local/cuda

结果没用,才发现这里面根本没有该文件,libcupti.so在文件夹/local/cuda/extras/CUPTI/lib64下,于是按照上面第一句话的格式添加了环境变量,然后 python mnist_with_summaries.py运行成功!
但是,在pycharm里面仍然不行,解决方法是:

在pycharm 选择 run->Edit Configurations->Enviroment Variables, 添加环境变量如下:

LD_LIBRARY_PATH  

/local/cuda/extras/CUPTI/lib64

运行成功!
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: