安装 Tensorflow 遇到的坑

2023年12月5日 208次阅读 来源: 阿o醒

现在 TF 官网上是1.5版本，需要用到 cuda-9.0 。
右上角切换到1.4版本，或者

sudo pip3 install tensorflow-gpu==1.4 # python3 大法好！

就可以在在 cuda-8.0 上使用了。
如果装完还是在 import tensorflow as tf时报

ImportError: libcublas.so.8.0: cannot open shared object file: No such file or directory

的错，则需要ldconfig来更新下库信息。

sudo ldconfig /usr/local/cuda-8.0/lib64

或者

# 未尝试
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-8.0/lib64
# 如果仍然不行，再尝试执行: 
export PATH=$PATH:/usr/local/cuda-8.0/bin 
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-8.0/lib64 
source /etc/profile
    原文作者：阿o醒
    原文地址: https://www.jianshu.com/p/4ff62bb27770
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞