2

我采取的步骤如下。在此./configure步骤中，除非我允许 cuDNN 版本 (v5) 的系统默认值而不是指定 v5.1.5（如我所愿），否则我会收到一条错误消息，指出 cuDNN 的环境版本 (v5) 与目标版本不匹配(5.1.5)。由于我从 Nvidia 下载了 CUDA 8.0 的 cuDNN v5.1.5 文件，我感到很困惑。

更重要的是，在第 240 行附近挖掘https://github.com/tensorflow/tensorflow/blob/master/third_party/gpus/cuda_configure.bzl之后，我看到 bazel 配置脚本检查的版本似乎只需要考虑到 cuDNN 版本的主要数字（我在 cudnn.h 中的主要、次要和补丁数字分别是 5、1、5）。所以，我不确定它如何允许安装 5.1.5 版。也许我看错了？

无论如何，有没有人可以在 g2.2xlarge 实例上安装带有 CUDA 8.0 和 Tensorflow 0.12 的 cuDNN 5.1.5 的方法？

谢谢！

脚步

（注意：这些工作，但它安装 cuDNN 5.0，而不是所需的 5.1.5）

供应

  • 遵循以下配置步骤：https ://medium.com/@giltamari/tensorflow-getting-started-gpu-installation-on-ec2-9b9915d95d6f#.2hv67eeek （即最多但不包括sudo apt-get update && sudo apt-get -y upgrade：）

安装依赖项和工具

安装 Cuda 8

  • wget https://developer.nvidia.com/compute/cuda/8.0/prod/local_installers/cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64-deb
  • sudo dpkg -i cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64-deb
  • rm cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64-deb
  • sudo apt-get update
  • sudo apt-get install -y cuda

安装 cuDNN

  • 我们要下载并安装最新版本的 cuDNN。下载 cuDNN 需要登录 NVIDIA 开发者网站，所以我们不能使用 wget 来获取文件。从 NVIDIA 下载以下文件并将它们上传到您的 AWS 实例。
  • 在 Linux 上为 CUDA 8.0 下载 cuDNN 5.1
  • scp -i ssh-key.pem path/to/downloaded/cudnn ubuntu@ec2{instance}.us-west-1.compute.amazonaws.com:~/
  • sudo tar -xzvf cudnn-8.0-linux-x64-v5.1.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*

配置环境

  • 将以下内容放在末尾~/.bashrc： export CUDA_HOME=/usr/local/cuda export CUDA_ROOT=/usr/local/cuda export PATH=$PATH:$CUDA_ROOT/bin export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CUDA_ROOT/lib64:$CUDA_ROOT/extras/CUPTI/lib64
  • source ~/.bashrc
  • sudo reboot

安装 Bazel

  • sudo add-apt-repository -y ppa:webupd8team/java
  • sudo apt-get update
  • echo debconf shared/accepted-oracle-license-v1-1 select true | sudo debconf- set-selections
  • echo debconf shared/accepted-oracle-license-v1-1 seen true | sudo debconf- set-selections
  • sudo apt-get install -y oracle-java8-installer
  • sudo apt-get install pkg-config zip g++ zlib1g-dev
  • scphttps://github.com/bazelbuild/bazel/releases/download/0.3.2/bazel-0.3.2-installer-linux-x86_64.sh从本地机器到 ec2 实例
  • chmod +x bazel-0.1.4-installer-linux-x86_64.sh
  • ./bazel-0.1.4-installer-linux-x86_64.sh --user
  • rm bazel-0.1.4-installer-linux-x86_64.sh
  • bazel version

构建和安装 TensorFlow

  • git clone --recurse-submodules https://github.com/tensorflow/tensorflow
  • cd tensorflow
  • TF_UNOFFICIAL_SETTING=1 ./configure
  • 全部按回车/默认，除了：
    • CUDA 版本 = 8.0，CUDA 依赖项 = 3.0 (k520 gpu)
  • bazel build -c opt --config=cuda //tensorflow/cc:tutorials_example_trainer
  • bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg
  • sudo pip install --upgrade /tmp/tensorflow_pkg/tensorflow-0.12.0rc1-cp27-cp27mu-linux_x86_64.whl
4

0 回答 0