导入tensorflow-gpu失败

来源:1-9 Google_cloud_gpu_tensorflow配置

慕虎6032090

2020-08-10

配置谷歌云GPU时,安装2.0.0alpha0的tensorflow-gpu成功,进入python后无法导入,失败原因是:无法加载native tensorflow runtime。查了一些解决办法,cd ..退回上一层无法解决;cuda版本是10.1,是按tensorflow官网提供的脚本进行,应该也不是版本号的问题;python3.6.9,不到3.7,应该也不是python问题。
http://img.mukewang.com/szimg/5f315fbd08de760906020392.jpg

http://img.mukewang.com/szimg/5f315fc4084d4b4706020411.jpg

http://img.mukewang.com/szimg/5f315fd208f5eb6106020219.jpg

写回答

1回答

正十七

2020-08-15

问题就在于找不到具体的一个so文件了,应该是版本不匹配的原因, alpha版需要的可能是10.0版本而你装的是10.1。可以试试如下方法,

去cuda的安装目录下,一般是/usr/local/cuda-10.1, 然后找到类似的文件,libcublas.so.10.1 , 然后用软链接创建一个对这个文件的引用,ln -s  libcublas.so.10.1 libcublas.so.10.0, 然后再运行下tensorflow。

如果不成,你可以试试不装alpha版,装正式版tensorflow-2.1或者2.2 。

0
0

Google老师亲授 TensorFlow2.0 入门到进阶

Tensorflow2.0实战—以实战促理论的方式学习深度学习

1849 学习 · 896 问题

查看课程