linux下tensorflow与cuda版本不兼容问题,ImportError: libcublas.so.10.0: cannot open shared object file

linux下tensorflow与cuda版本不兼容问题


因为跑一个多目标跟踪的代码,要用到yolov3,所以作为pytorch用户的我,第一次用上了tensorflow。
但是我使用的镜像环境,是cuda10.1版本的:
在这里插入图片描述
代码使用的tensorflow,是1.13.1版本的;需要将cuda降到10.0。
这里有一篇 博客,展示了Tensorflow不同版本要求与CUDA及CUDNN版本对应关系,一般我们都是装好了tensorflow-gpu,然后需要对应版本的CUDA及CUDNN,可以让大家参考一下。

用conda安装对应的cudatoolkit

conda install cudatoolkit=10.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/

用conda安装对应的cudnn

conda install cudnn=7.5.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/

这样每次运行时,会链接到这个版本的cuda和cudnn,实际的cuda版本没有变。

参考链接:
tf-gpu报错:ImportError: libcublas.so.10.0: cannot open shared object file: No such file or directory
Ubuntu下使用conda在虚拟环境中安装CUDA、CUDNN及Tensorflow
conda安装cudnn和cudatoolkit

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值