问题解决笔记 -- Ubuntu 安装 tensorflow与问题解决笔记

来源:互联网 发布:如何建设企业文化 知乎 编辑:程序博客网 时间:2024/06/06 00:17

引用的文章

1.以Python 2.7 ,Ubuntu14.04为基础环境

# Ubuntu/Linux 64-bit, CPU only, Python 2.7:$ sudo pip install --upgrade https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-0.8.0-cp27-none-linux_x86_64.whl# Ubuntu/Linux 64-bit, GPU enabled, Python 2.7. Requires CUDA toolkit 7.5 and CuDNN v4. 使用GPU加速,需安装CUDA及CuDNN# For other versions, see "Install from sources" below.$ sudo pip install --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.8.0-cp27-none-linux_x86_64.whl

笔者 使用的tensorflow包是包含GPU enabled的,所以以下是安装过程中出现问题的相应笔记

执行相应的tensorflow安装命令后,开始安装CUDA及CuDNN
下载并安装 Cuda Toolkit 7.5或8.0
https://developer.nvidia.com/cuda-toolkit-archive

选择deb(local)包或deb(network)包

将包传至Ubuntu,到相应目录下执行以下命令

sudo dpkg -i 下载后相应的deb包名.debsudo apt-get updatesudo apt-get install cuda

等待完成安装

下载并安装 CUDNN Toolkit 6.5
百度搜索相应的包 cudnn-6.5-linux-x64-v2.tgz 下载

解压并拷贝 CUDNN 文件到 Cuda Toolkit 安装路径下. 假设 Cuda Toolkit 安装 在 /usr/local/cuda, 执行以下命令:

tar xvzf cudnn-6.5-linux-x64-v2.tgzsudo cp cudnn-6.5-linux-x64-v2/cudnn.h /usr/local/cuda/includesudo cp cudnn-6.5-linux-x64-v2/libcudnn* /usr/local/cuda/lib64

sudo vim ~/.bashrc 打开“./bashrc”文件,然后
在打开的文件的末尾添加如下代码并保存:

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64"export CUDA_HOME=/usr/local/cuda

注: cuda默认安装在/usr/local/cuda文件夹下

2 . 错误处理
输入tensorflow测试代码后:

若出现了libcudart.so.7.5: cannot open shared object file: No such file or directory的错误:
解决方法,以cuda-8.0 为例
由于是版本问题,于是建立相应的软链接

ln -s /usr/local/cuda-8.0/lib64/libcudart.so.8.0 /usr/lib ln -s /usr/local/cuda-8.0/lib64/libcudart.so.8.0 /usr/lib/libcudart.so.7.5sudo ldconfig

若出现相应的CUDNN错误,则可能是CUDNN未安装,按照上面的步骤进行拷贝
3 . tensorflow测试代码

$ python>>> import tensorflow as tf>>> hello = tf.constant('Hello, TensorFlow!')>>> sess = tf.Session()>>> print sess.run(hello)Hello, TensorFlow!>>> a = tf.constant(10)>>> b = tf.constant(32)>>> print sess.run(a+b)42

4 . 其他操作
4.1 要是升级相应的CUDNN文件后,要删除旧版本释放空间,使用命令

apt-get autoremove

4.2 假如要在/usr/local目录下建立软链接,如使得cuda链接到cuda-8.0
可执行以下命令(慎用)

# rm -rf cuda# ln -s cuda-8.0 cuda# ldconfig /usr/local/cuda/lib64

记得加上ldconfig,否则可能导致错误 error while loading shared libraries

4.3 删除tensorflow

sudo pip uninstall tensorflow

4.4 删除cuda
到相应的版本目录下,以7.5为例

sudo /usr/local/cuda-7.5/bin/uninstall_cuda-7.5.pl

如果没有相应的文件 ,执行以下命令

sudo apt-get purge cudasudo apt-get autoremove

4.5 删除显示驱动

sudo /usr/bin/nvidia-uninstall

如果找不到,用命令

sudo apt-get install autoremove --purge nvidia*
0 0
原创粉丝点击