ubuntu16.04下在已配好cuda-8.0和cudnn的基础上安装tensorflow0.8.0踩过的坑

在Ubuntu16.04上,已经安装了CUDA-8.0和CUDNN的基础上尝试使用pip安装tensorflow0.8.0时遇到了问题。首先,由于pip版本导致的错误,通过升级pip解决了。接着,导入tensorflow时报错找不到libcudart.so.7.5,通过软链接将其指向CUDA8.0的相应库文件。之后,运行tensorflow的简单示例时又遇到错误,经过查找资料发现需要安装libtcmalloc-minimal4,并修改.bashrc文件配置环境变量。最后,重启终端,所有问题得到解决,成功运行tensorflow程序。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

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"

同时必须关闭终端,然后再打开才能生效


至此成功!




评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值