1.采用pip直接安装:打开终端,直接输入命令:
sudo pip install --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.8.0-cp27-none-linux_x86_64.whl
安装的过程中可能会报错,因为pip的版本问题,可能需要升级pip,这里直接根据它的报错提示进行操作即可,问题不大
2.此时,在终端输入python,然后import tensorflow,我出现了一个错误:libcudart.so.7.5: cannot open shared object file: No such file or directory。对于这个错误我们需要用ln将需要的so文件链接到/usr/lib或者/lib这两个默认的目录下边
在这里我用的
sudo ln -s /usr/local/cuda-8.0/lib64/libcudart.so.8.0 /usr/lib
和
sudo ln -s /usr/local/cuda-8.0/lib64/libcudart.so.8.0 /usr/lib/libcudart.so.7.5
然后
sudo lgconfig
这时import tensorflow成功
3.接下来,我开始执行官网第一个程序
import tensorflow as tf
matrix1 = tf.constant([[3., 3.]])
matrix2 = tf.constant([[2.],[2.]])
product = tf.matmul(matrix1, matrix2)
sess = tf.Session()
result = sess.run(product)
print result
sess.close()
程序执行到result = sess.run(product)这里时就报错
这里我搞了一天,查看了很多博客,都没找到解决办法,这时我看了一篇stackoverflow的一篇文章的解答
这里说可能需要libtcmalloc-minimal4,执行以下命令下载安装
sudo apt-get install libtcmalloc-minimal4
同时并打开打开~/.bashrc文件
sudo gedit ~/.bashrc
并将以下内容写入到~/.bashrc文件中(配置环境变量)
export LD_PRELOAD="/usr/lib/libtcmalloc_minimal.so.4"
同时必须关闭终端,然后再打开才能生效
至此成功!