导入tensorflow:ImportError: libcublas.so.9.0 报错

yipeiwu_com7年前Python基础

错误: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

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持【听图阁-专注于Python设计】。

相关文章

使用python进行拆分大文件的方法

使用python进行拆分大文件的方法

python按指定行数把大文件进行拆分 如图大文件有7000多万行,大小为16G 需要拆分成多个200万行的小文件 代码如下: # -*- coding:utf-8 -*- fro...

Python集中化管理平台Ansible介绍与YAML简介

Python集中化管理平台Ansible介绍与YAML简介

本文实例讲述了Python集中化管理平台Ansible介绍与YAML。分享给大家供大家参考,具体如下: 一 中文社区 http://ansible.cn/forum.php 二 点睛 A...

python数据挖掘需要学的内容

1、Pandas库的操作 Panda是数据分析特别重要的一个库,我们要掌握以下三点: · pandas 分组计算; · pandas 索引与多重索引; 索引比较难,但是却是非常重要的 ·...

Python实现代码统计工具

本文实例为大家分享了Python实现代码统计工具的具体代码,供大家参考,具体内容如下 思路:首先获取所有文件,然后统计每个文件中代码的行数,最后将行数相加. 实现的功能: 统计每个文件的...

pytorch AvgPool2d函数使用详解

我就废话不多说了,直接上代码吧! import torch import torch.nn as nn import torch.nn.functional as F from to...