问题:
今天在安装imgaug数据增强库的时候,提示TensorFlow需要1.7.0版本

升级tensorflow版本之后,import tensorflow的时候出现如下错误:
ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory

原因:
出现该错误的主要原因:cuda未安装或者cuda的版本有问题
即新升级的tensorflew(1.7.0)版本与我原来的CUDA(8.0.61)版本不匹配
这里附上linux查询CUDA和CUDNN 版本的方法:
1)查询CUDA版本方法:
输入:
cat /usr/local/cuda/version.txt

可以看到我的CUDA版本是8.

升级TensorFlow到1.7.0后,遇到ImportError,原因是CUDA版本不匹配。解决方案包括降级TensorFlow到1.4.0或安装CUDA 9.0,并更新环境变量。文中还提供了查询CUDA和CUDNN版本的方法。
最低0.47元/天 解锁文章
1062

被折叠的 条评论
为什么被折叠?



