安装 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 
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容