安装环境:Ubuntu 16.04lts 64位,gcc5.4
1.安装Cuda
1. 下载cuda toolkit。
下载cuda8.0 地址:https://developer.nvidia.com/cuda-downloads 版本选择:Operating System:Linux; Architecture : x86_64; Distribution : Ubuntu; Version : 16.04; Installer Type : runfile(local)。只下载Base Installer就可以了。
2. 禁用 nouveau
终端运行:
lsmod | grep nouveau
若有输出则代表nouveau正在加载。
禁用方法:在/etc/modprobe.d中创建文件blacklist-nouveau.conf,在文件中输入以下两行内容
blacklist nouveau options nouveau modeset=0
终端输入
sudo update-initramfs –u
设置完毕可以再次运行 lsmod | grep nouveau查看nouveau是否正在加载。如果仍输出信息,则重启电脑。没有输出信息则代表禁用成功。
3. 安装cuda toolkit
7. Configure the installation
切换目录到tensorflow安装目录下,会有一个configure文件,输入命令
./configure
8. Build the pip package
输入命令
bazel build --config=opt --config=cuda --cxxopt="-D_GLIBCXX_USE_CXX11_ABI=0" //tensorflow/tools/pip_package:build_pip_package
命令 --cxxopt="-D_GLIBCXX_USE_CXX11_ABI=0" 是为了gcc兼容性。
bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg
上述命令执行完毕后,在/tmp/tensorflow_pkg
directory目录下会有.whl
类型的文件,执行命令:
9.Install the pip package
执行命令,注意更换文件名:
sudo pip install /tmp/tensorflow_pkg/tensorflow-1.3.0-py2-none-any.whl