在虚拟环境pytho3.6、cuda10.2中使用pip安装tensorflow-gpu==1.14运行报错,ImportError: libcublas.so.10.0: cannot open shared object file: No such file or directory,查找了相关资料说是tensorflow和cuda版本不匹配问题,但是我已经安装了cuda10.2重新安装比较麻烦。

卸载环境中pip安装的tensorflow-gpu,使用conda安装后程序运行正常,完美解决。

conda activate darknet
pip uninstall tensorflow-gpu
conda install tensorflow-gpu==1.14.0

参考资料:https://www.zhihu.com/question/365667275

Logo

昇腾计算产业是基于昇腾系列(HUAWEI Ascend)处理器和基础软件构建的全栈 AI计算基础设施、行业应用及服务,https://devpress.csdn.net/organization/setting/general/146749包括昇腾系列处理器、系列硬件、CANN、AI计算框架、应用使能、开发工具链、管理运维工具、行业应用及服务等全产业链

更多推荐