ubuntu14.04安装tensorflow+cuda8.0+cudnn5.1官方教程

来源:互联网 发布:java处理xml 编辑:程序博客网 时间:2024/05/22 12:56

一、安装ubuntu14.04

      如果想做双系统,可以考虑将硬盘分出一个区,删除分区后变成空闲。将ubuntu系统安装在空闲存储空间中,有管ubuntu系统安装,详情http://jingyan.baidu.com/article/76a7e409bea83efc3b6e1507.html

二、安装什么的TensorFlow的版本,CPU版本?、GPU?(GTX 1050)

        参考地址:Installing TensorFlow on Ubuntu

        TensorFlow with CPU support only,改版本仅有CPU情况下安装,安装过程比较方便,仅5-10分钟。具体步骤如下:

        方法一:传统的pip安装

依赖包

sudo apt-get install python-pip python-dev
sudo pip install tensorflow  or sudo pip install --upgrade TF_PYTHON_URL
TF_PYTHON_URL表示TensorFlow的下载链接,因为pip可能找不到TensorFlow的来源,所以我们直接给定它的源,cpu版本的源如下
https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.0.1-cp27-none-linux_x86_64.whl
gpu的源

https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.0.1-cp27-none-linux_x86_64.whl
如果用python3.4版本安装,

sudo pip3 install tensorflow
或者
sudo pip3 install --upgrade TF_PYTHON_URL

TF_PYTHON_URL链接如下:

https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.0.1-cp34-cp34m-linux_x86_64.whl
       方法二:Installing with virtualenv
sudo apt-get install python-pip python-dev python-virtualenv
virtualenv --system-site-packages ~/tensorflow
source ~/tensorflow/bin/activate
source ~/tensorflow/bin/activate.csh

三、cuda-8.0+cudnn5.1

        根据GPU的型号,下载相应的cuda版本。cuda版本的下载地址https://developer.nvidia.com/cuda-toolkit-archive

验证你的显卡是否支持cuda

$ lspci | grep -i nvidia

接着下载对应的cuda版本,选择deb包,如cuda-repo-ubuntu1404_8.0-14_amd64.deb。cd到下载目录

sudo dpkg -i cuda-repo-<distro>_<version>_<architecture>.deb
sudo apt-get update
开始安装cuda以及显卡驱动,安装cuda的同时就会把显卡驱动也全部安装,所以建议装完ubuntu以后直接安装cuda,不需要专门下载显卡驱动安装,再装cuda。
sudo apt-get install cuda
配置环境变量,并将以下两行添加到 隐藏文件~/.bashrc 的最后一行。

sudo gedit ~/.bashrc

export PATH=/usr/local/cuda/bin:$PATH  export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
验证cuda是否装好

查看显卡的驱动版本,查看nvcc编译器的版本

$ nvidia-smi$ cuda-install-samples-8.0.sh <dir>$ cat /proc/driver/nvidia/version$ nvcc -V i
编译cuda的示例代码:然后make一下编译代码。

$ cd ~/NVIDIA_CUDA-8.0_Samples$ make
直接cd进入bin路径运行devicequery

$ cd bin$ ./deviceQuery
最后有显示pass,则表示成功。
cudnn安装步骤:
1、从官网上下载cudnn的安装包。
2、将安装包解压,将此安装包放在home路径下即可,并在当前路径下进行解压,解压后的文件夹名为cuda。

3、在终端上编辑如下代码:

$cd cuda/include
$sudo cp *.h /usr/local/include/
$cd ../lib64
$sudo cp libcudnn* /usr/local/cuda/lib64/
$ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

四、安装python的交互式界面spyder

        先安装依赖库

sudo apt-get install python-qt4 python-sphinxsudo pip install spyder


0 0