Ubuntu16.04 安装CUDA8.0+CUDNN6.0+Tensorflow-GPU1.4版本出现问题解决方案

版权声明:个人原创,转载请申明 https://blog.csdn.net/baidu_38172402/article/details/86668618

问题一:ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory

安装的Tensorflow-GPU版本过高,使用命令:sudo apt-get install tensorflow-gpu,默认安装的是当前发行的最高版本。

正确的命令是:sudo pip install tensorflow-gpu==1.4.0 (直接指明版本号)。

问题二:ImportError: numpy.core.multiarray failed to import

安装的Numpy版本过高,类似问题一,所以正确的命令是:sudo pip install numpy==1.15.4

安装注意事项,注意当前的python版本,一般为python2.7和python3.5,注意可用sudo update-alternatives --config python 进行版本选择(前提是设置过),同时要注意CUDA+CUDNN+Tensorflow-GPU的版本搭配。

pip list //可查看当前python版本安装的所有库,其库目录为/usr/local/lib/python2.7/dist-packages/。

sudo pip uninstall  libname(可由pip list查出)  //对相应的库进行卸载。

watch -n 1 nvidia-smi //可以实时监控GPU使用情况。

猜你喜欢

转载自blog.csdn.net/baidu_38172402/article/details/86668618