登录 注册

如何在 Ubuntu 下编译 CUDA 10.0 TensorFlow-GPU r.1.13 版本

本贴最后更新于 1851 天前，其中的信息可能已经水流花落

如何在 Ubuntu 下编译 CUDA 10.0 TensorFlow-GPU r.1.13 版本

注意:

  1. CUDA 10.1 版本在编译时会报错，提示版本无法适配 kernel。可以使用 mazar 的方法解决部分问题。但是使用 CUDA 10.0 + cuDNN 7.5.0 会容易很多

    The driver installation is unable to locate the kernel source. Please make sure that the kernel source packages are installed and set up correctly.

  2. bazel 的版本建议是 0.15~0.17，最高不要超过 0.21，会导致无法编译。

  3. 该重启的时候需要重启。

  4. 编译会花较长时间(3~4H)，建议再空闲时间编译。

  5. 请严格按照此教程的顺序进行安装和编译~~~

这是一个如何编译适配 CUDA 10.0 版本 TensorFlow GPU r1.13 的教程。(截至 2019 年 03 月 26 日 12:00)由于最新的官方发布中，TensorFlow 的版本为 1.12，使用 cuda 9.0 + cuDNN 7。最新编译同样如此。但从 TensorFlow 的项目 issues 来看，最新 r.1.13 是适配 CUDA 10.0 的，而 CUDA 每次大版本的更新都有很大的性能上的提升。由于 TensorFlow 的 github release 只有源码，因此我们考虑从源码编译最新版的 TensorFlow。

最新版本的 TensorFlow 提供了最新的特性和优化方法，最新的 CUDA Toolkit 能提供速度上的提升，也能使用最新 cuDNN 减少深度学习训练时间。

Step 1. 更新系统

 sudo apt-get update   
 sudo apt-get upgrade

Step 2. 确定 linux 的版本，安装最新的依赖

 uname -m && cat /etc/*release  
 sudo apt-get install build-essential   
 sudo apt-get install cmake git unzip zip  
 # 这边更建议下载最新的conda进行安装

同时使用 Conda 安装 Python3。可以参考 Conda 加速下载。可以下载最新安装包之后 conda install python=3.6，也可以创建虚拟环境安装 conda create -n env_name python=3.6

Step 3. 安装 linux kernel header

 sudo apt-get install linux-headers-$(uname -r)

安装好之后，可以查看其它内核版本是否已经安装在 boot 目录下。

如果出现如下类似问题，那可能是 boot 目录下空间不够

 gzip: stdout: No space left on device  
 update-initramfs: failed for /boot/initrd.img-4.4-143-generic with 1  
 E: Sub-process /usr/bin/dpkg returned an error code (1)

采用如下方案解决:

 sudo dpkg --configure -a  
 dpkg -l | grep linux-  
 # 把和uname -r不一致的旧内核都可以删了  
 sudo dpkg --purge linux-headers-x.x.x-xx-blabla  
 sudo apt autoremove

比如我的 uname -r 是 4.4.0-143-generic, 把列表中其他 header 和 image 删除就好了。

Step 4. 确定 GPU 适配的 CUDA 版本

查看官网的 cuda-gpus，由于安装新版的 CUDA，建议移除旧版 CUDA

 lspci | grep -i nvidia  
 sudo apt-get purge nvidia*  
 sudo apt-get autoremove  
 sudo apt-get autoclean  
 sudo rm -rf /usr/local/cuda*

接下来安装 cuda 和驱动：

 sudo apt-get update   
 sudo apt-get -o Dpkg::Options::="--force-overwrite" install cuda-10-0 cuda-drivers

也可以使用 deb 进行安装。

Step 5.安装后按提示重启。

Step 6.更新 ~/.bashrc

首先更新~/.bashrc 中的环境变量

 echo 'export CUDA_HOME=/usr/local/cuda-10.0'  
 echo 'export PATH=$CUDA_HOME/bin${PATH:+:${PATH}}' >> ~/.bashrc  
 echo 'export LD_LIBRARY_PATH=$CUDA_HOME/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}' >> ~/.bashrc  
 ## 或者直接编辑~/.bashrc  
 ​  
 source ~/.bashrc  
 sudo ldconfig

之后查看 nvidia-smi，确认 cuda 版本和驱动版本。比如我的是 NVIDIA-SMI 418.39 Driver Version: 418.39 CUDA Version: 10.0。之后到 nvidia-samples 验证是否安装成功。

 # sample 可以指定路径安装  
 cuda-install-samples-10.0.sh ~  
 cd ~/NVIDIA_CUDA-10.0_Samples/5_Simulations/nbody  
 ​  
 # 也可能在安装cuda的过程中安装，路径位置 /usr/local/cuda-10.0/samples 再复制到指定位置即可  
 # cp -r /usr/local/cuda-10.0/samples ~/NVIDIA_CUDA-10.0_Samples  
 ​  
 make ./nbody

不报错就成功了。

Step 7. 安装 cuDNN。

打开 cuDNN 网站进行下载，选择 cuDNN Library for Linux。下载之后是 .tgz 压缩包。

 tar -xf cudnn-10.0-linux-x64-v7.5.0.56.tgz  
 ​  
 sudo cp -R cuda/include/* /usr/local/cuda-10.0/include  
 ​  
 sudo cp -R cuda/lib64/* /usr/local/cuda-10.0/lib64

查看 CUDA 和 cuDNN 版本验证一下

 cat /usr/local/cuda/version.txt  
 cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

Step 8. 安装新版 NCCL

这里需要下载对应系统版本的 nccl，例如 ubuntu 16.04 下载Local installer for Ubuntu 16.04

参考 nccl

 # 或者  
 sudo apt install libnccl2=2.4.2-1+cuda10.0 libnccl-dev=2.4.2-1+cuda10.0  
 sudo ldconfig

Step 9. 更新 python 依赖

建议在虚拟环境中

 pip install -U pip six numpy wheel mock  
 pip install -U keras_applications==1.0.5 --no-deps  
 pip install -U keras_preprocessing==1.0.3 --no-deps

Step 10. 配置编译

下载安装 bazel：

 wget -c https://github.com/bazelbuild/bazel/releases/download/0.21.0/bazel-0.21.0-installer-linux-x86_64.sh  
 chmod +x bazel-0.21.0-installer-linux-x86_64.sh  
 ./bazel-0.21.0-installer-linux-x86_64.sh --user  
 # which bazel; whereis bazel 查看一下路径  
 # 看是否需要加到环境配置中。  
 echo 'export PATH="$PATH:$HOME/bin"' >> ~/.bashrc  
 ​  
 # 重新加载环境变量  
 source ~/.bashrc  
 sudo ldconfig

下载 TensorFlow 源码:

 cd ~  
 git clone https://github.com/tensorflow/tensorflow.git  
 cd tensorflow  
 git checkout r1.13  
 ./configure

配置编译 config，有默认项的可以直接使用默认项，除非像 cuda 和 python path 之类的指定版本。

 # 最好指定python3.6的路径，比如我的是虚拟环境/home/dxm/data/anaconda2/envs/py36torch  
 Please specify the location of python. [Default is /usr/bin/python]: /home/dxm/data/anaconda2/envs/py36torch/bin/python3  
 ​  
 # 配置其他选项  
 Do you wish to build TensorFlow with Apache Ignite support? [Y/n]: Y  
 Do you wish to build TensorFlow with XLA JIT support? [Y/n]: Y  
 Do you wish to build TensorFlow with OpenCL SYCL support? [y/N]: N  
 Do you wish to build TensorFlow with ROCm support? [y/N]: N  
 Do you wish to build TensorFlow with CUDA support? [y/N]: Y  
 Please specify the CUDA SDK version you want to use. [Leave empty to default to CUDA 9.0]: 10.0  
 Please specify the location where CUDA 10.0 toolkit is installed. Refer to README.md for more details. [Default is /usr/local/cuda]: /usr/local/cuda-10.0  
 Please specify the cuDNN version you want to use. [Leave empty to default to cuDNN 7]: 7.5.0  
 ​  
 Please specify the location where cuDNN 7 library is installed. Refer to README.md for more details. [Default is /usr/local/cuda-10.0]: /usr/local/cuda-10.0  
 Do you wish to build TensorFlow with TensorRT support? [y/N]: N  
 Please specify the NCCL version you want to use. If NCCL 2.2 is not installed, then you can use version 1.3 that can be fetched automatically but it may have worse performance with multiple GPUs. [Default is 2.2]: 2.4.2  
 ​  
 Do you want to use clang as CUDA compiler? [y/N]: N  
 ​  
 Please specify which gcc should be used by nvcc as the host compiler. [Default is /usr/bin/gcc]: /usr/bin/gcc  
 ​  
 Do you wish to build TensorFlow with MPI support? [y/N]: N  
 ​  
 Please specify optimization flags to use during compilation when bazel option "--config=opt" is specified [Default is -march=native]: -march=native  
 ​  
 Would you like to interactively configure ./WORKSPACE for Android builds? [y/N]:N

Step 11 使用 bazel 编译 TensorFlow

 bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package

接下来，等着吧。等几个小时，睡觉或者打游戏去吧。

编译好了之后，构建 wheel 文件

 bazel-bin/tensorflow/tools/pip_package/build_pip_package tensorflow_pkg  
 cd tensorflow_pkg  
 tensorflow-1.13.1-cp36-cp36m-linux_x86_64.whl

使用 tensorflow-1.13.1-cp36-cp36m-linux_x86_64.whl 安装在需要的环境就 OK 啦。

接下来验证一下是否安装成功。

 import tensorflow as tf  
 hello = tf.constant('Hello, TensorFlow!')  
 sess = tf.Session()  
 print(sess.run(hello))  
 # b'Hello, TensorFlow!'  
 print(tf.__version__)  
 # 1.13.1
  • TensorFlow

    TensorFlow 是一个采用数据流图（data flow graphs），用于数值计算的开源软件库。节点（Nodes）在图中表示数学操作，图中的线（edges）则表示在节点间相互联系的多维数据数组，即张量（tensor）。

    20 引用 • 19 回帖
  • 编译
    17 引用 • 22 回帖
  • cuda
    2 引用
  • bazel
    1 引用
1.8k 10 759 81 177 51 9 62 618

相关帖子

回帖
如何在 Ubuntu 下编译 CUDA 10.0 TensorFlow-GPU r.1.13 版本

欢迎来到这里！

我们正在构建一个小众社区，大家在这里相互信任，以平等 • 自由 • 奔放的价值观进行分享交流。最终，希望大家能够找到与自己志同道合的伙伴，共同成长。

注册 关于
请输入回帖内容 ...
lai-bluejay
回帖
13
 帖子
13
 积分
607