Ubuntu 16.04下安装Tensorflow(GPU)
Posted alan215m
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Ubuntu 16.04下安装Tensorflow(GPU)相关的知识,希望对你有一定的参考价值。
参考:http://blog.sina.com.cn/s/blog_672f698e0102wavp.html
1.首先安装nvidia显卡驱动:
系统设置->软件更新->附加驱动->选择nvidia最新驱动(361)->应用更改
2.下载CUDA8.0 地址https://developer.nvidia.com/cuda-release-candidate-download(需要登陆)
3.安装cuda
sudo dpkg -i cuda-repo-ubuntu1604-8-0-rc_8.0.27-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda
4.下载cudnn,最好是v4版本 地址https:developer.nvidia.com/cudnn
5.把cudnn解压复制到cuda下
tar xvzf
cudnn-7.0-linux-x64-v4.0-prod.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
6.从源码安装tensorflow
3、ubuntu的gcc编译器是5.4.0,然而cuda8.0不支持5.0以上的编译器,因此需要降级,把编译器版本降到4.9:
- sudo apt-get install g++-4.9
- sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.9 20
- sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-5 10
- sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-4.9 20
- sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-5 10
- sudo update-alternatives --install /usr/bin/cc cc /usr/bin/gcc 30
- sudo update-alternatives --set cc /usr/bin/gcc
- sudo update-alternatives --install /usr/bin/c++ c++ /usr/bin/g++ 30
- sudo update-alternatives --set c++ /usr/bin/g++
git clone https://github.com/tensorflow/tensorflow
进入tensorflow根目录(home下)运行./configure配置tensorflow
7.安装Bazel(最好下好了再安装,第二种方法,不然需要FQ)
http://www.bazel.io/docs/install.html
安装其他依赖
sudo apt-get install python-numpy swig python-dev python-wheel
8.建立GPU支持
bazel build -c opt --config=cuda //tensorflow/cc:tutorials_example_trainer
如发生错误可在后面加上--verbose_failures运行如下
bazel build -c opt --config=cuda //tensorflow/cc:tutorials_example_trainer
--verbose_failures
完成后运行
bazel-bin/tensorflow/cc/tutorials_example_trainer --use_gpu
此程序会计算一个2*2矩阵的主特征值 输出如下
创建PIP安装包
bazel build -c opt --config=cuda //tensorflow/tools/pip_package:build_pip_package
--config=cuda表示支持gpu
bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg
sudo pip install /tmp/tensorflow_pkg/tensorflow-0.9.0-py2-none-any.whl
OK,搞完收工
以上是关于Ubuntu 16.04下安装Tensorflow(GPU)的主要内容,如果未能解决你的问题,请参考以下文章
Ubuntu16.04下安装tensorflow(Anaconda3+pycharm+tensorflow+CPU)
Ubuntu 16.04下安装Tensorflow(GPU)
ubuntu16.04下安装TensorFlow(GPU加速)----详细图文教程
ubuntu16.04下安装CUDA,cuDNN及tensorflow-gpu版本过程