作者:looloSam | 来源:互联网 | 2023-10-10 20:00
以前就听说过tensorflow-GPU版的要注意版本,今天在GPU服务器上安装tensorflow后可算是见识到了,各种报错啊,简直心累!!???良心填坑???
今天直接在一台装有cuda9.0+cudnn-v7的GPU服务器里安装tensorflow,就是直接pip3安装的,然后import tensorflow这一步,就报错了:
ImportError: libcublas.so.10.0: cannot open shared object file: No such file or directory
一顿搜索啊,结合这些博客:https://blog.csdn.net/w5688414/article/details/79187499,https://blog.csdn.net/twt520ly/article/details/79415787,https://blog.csdn.net/qysh123/article/details/79977891),以及CUDA和cudnn这两者的对应关系如下(按照这篇博客的介绍https://blog.csdn.net/gsch_12/article/details/79368990):
我的应该是CUDA 9.0→cuDNN v7.0.5,然后呢,tensorflow 1.6/1.5和CUDA 9.0对应,于是乎,我要卸载安装好的tensorflow,降低下版本,安装个新的低版本的吧,
狠心删掉原来的tensorflow版本:
sudo pip uninstall tensorflow-gpu
我的天,原来我安装的是1.13,这版本也太高了吧,难怪一堆问题,,,,
然后安装1.5版本的tensorflow:
sudo pip3 install -U --pre tensorflow-gpu==1.5
然后就不报标题中的错了,真的费心费力,每次安装个东西,各种版本问题,希望以后不要再被版本困惑了~祝大家好运~