安装 Tensorflow 遇到的坑

现在 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 

你可能感兴趣的:(安装 Tensorflow 遇到的坑)