2017-06-24 4 views
0

私はTensorflowを自分でコンパイルしようとしました。(コンパイル済みのホイールを使用しても動作は遅くなりますが)私は./configureを使用して、cuDNNバージョン5.1.10を使用するように設定し明示的に指定します。その前に、私はcuDNN 5.1.10をダウンロードし、自分のファイルをCudaディレクトリにコピーしました。Tensorflowは、5.1.10が指定されていてもcuDNN 5.0でコンパイルします

build --action_env PYTHON_BIN_PATH="/home/ubuntu/project/venv/bin/python" 
build --action_env PYTHON_LIB_PATH="/home/ubuntu/project/venv/lib/python3.5/site-packages" 
build --define PYTHON_BIN_PATH="/home/ubuntu/project/venv/bin/python" 
build --define PYTHON_LIB_PATH="/home/ubuntu/project/venv/lib/python3.5/site-packages" 
build --force_python=py3 
build --host_force_python=py3 
build --python3_path="/home/ubuntu/project/venv/bin/python" 
test --force_python=py3 
test --host_force_python=py3 
test --define PYTHON_BIN_PATH="/home/ubuntu/project/venv/bin/python" 
test --define PYTHON_LIB_PATH="/home/ubuntu/project/venv/lib/python3.5/site-packages" 
run --define PYTHON_BIN_PATH="/home/ubuntu/project/venv/bin/python" 
run --define PYTHON_LIB_PATH="/home/ubuntu/project/venv/lib/python3.5/site-packages" 
build --define with_jemalloc=true 
build:opt --cxxopt=-march=native --copt=-march=native 
build --action_env TF_NEED_CUDA="1" 
build --action_env TF_NEED_OPENCL="0" 
build --action_env TF_CUDA_CLANG="0" 
build --action_env CUDA_TOOLKIT_PATH="/usr/local/cuda" 
build --action_env TF_CUDA_VERSION="8.0" 
build --action_env GCC_HOST_COMPILER_PATH="/usr/bin/gcc" 
build --action_env TF_CUDNN_VERSION="5.1.10" 
build --action_env CUDNN_INSTALL_PATH="/usr/local/cuda-8.0" 
build --action_env TF_CUDNN_VERSION="5.1.10" 
build --action_env TF_CUDA_COMPUTE_CAPABILITIES="3.7" 
build --config=cuda 
test --config=cuda 

お知らせTF_CUDNN_VERSION = "5.1.10" の部分:ここで

は、コンフィグ(.tf_configure.bazelrc)は次のようになります。次の私は、次のコマンドを実行します。

bazel build --cxxopt="-D_GLIBCXX_USE_CXX11_ABI=0" --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package 

bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg 

pip3 install --upgrade /tmp/tensorflow_pkg/tensorflow-1.2.0-cp35-cp35m-linux_x86_64.whl 

をそして最後に、私は何が起こっているか、次のエラーメッセージ

Loaded runtime CuDNN library: 5110 (compatibility version 5100) but source was compiled with 5005 (compatibility version 5000). If using a binary install, upgrade your CuDNN library to match. If building from sources, make sure the library loaded at runtime matches a compatible version specified during compile configuration. 

を取得しますか?何とかTensorflowは、正しいバージョンを指定しても、cuDNN 5.0を使用してコンパイルされますか?

システムはUbuntu 16、AWS p2インスタンスです。

答えて

0

私がこれを修正できる唯一の方法は、すべてをやり直すことでした(AWSマーケットプレイスのUbuntuの新しいバージョン)。

この疑問を抱いている戻って、問題を解決しようとするではなく、他のケースの誰かにはポイントが存在しないので、私はこの答えを受け入れるだろう、私はこの記事は非常に役に立った: https://developer.nvidia.com/compute/cuda/8.0/Prod2/local_installers/cuda_8.0.61_375.26_linux-run

関連する問題