首页 > 其他 > 详细

运行TensorFlow代码时报错

时间:2019-05-18 22:52:42      阅读:104      评论:0      收藏:0      [点我收藏+]

运行TensorFlow代码时报错

错误信息ImportError: libcublas.so.10.0: cannot open shared object file

原因:TensorFlow版本与CUDA版本不匹配

可使用pip3 install tensorflow-gpu==[version]将TensorFlow版本切换到制定版本中去,其对应的version可在‘>TensorFlow官网查看

另:

1.查看cuda版本(在ubuntu下):

cat /usr/local/cuda/version.txt
或者

nvcc --version

2.查看cudnn版本(在ubuntu下):

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
3.查看tensorflow-gpu的版本:

pip list

运行TensorFlow代码时报错

原文:https://www.cnblogs.com/youpeng/p/10887389.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!