Created
November 1, 2018 03:42
-
-
Save neofob/103e6a45d1346915405c9e1f6dab3886 to your computer and use it in GitHub Desktop.
tensorflow stuff
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
# Alternatively, having a symlink of /usr/local/cuda points to installed cuda version | |
export PATH=/usr/local/cuda-10.0/bin:$PATH | |
export CUDA_HOME=/usr/local/cuda |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
# with bazel 0.19, add tools/bazel.rc as below | |
import /tmp/tensorflow/tools/bazel.rc | |
import /tmp/tensorflow/.tf_configure.bazelrc |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
build --action_env PYTHON_BIN_PATH="/home/tuan/.virtualenvs/tf-build/bin/python" | |
build --action_env PYTHON_LIB_PATH="/home/tuan/.virtualenvs/tf-build/lib/python3.5/site-packages" | |
build --python_path="/home/tuan/.virtualenvs/tf-build/bin/python" | |
build:ignite --define with_ignite_support=true | |
build --define with_xla_support=true | |
build --action_env TF_NEED_OPENCL_SYCL="0" | |
build --action_env TF_NEED_ROCM="0" | |
build --action_env TF_NEED_CUDA="1" | |
build --action_env CUDA_TOOLKIT_PATH="/usr/local/cuda" | |
build --action_env TF_CUDA_VERSION="10.0" | |
build --action_env CUDNN_INSTALL_PATH="/usr/local/cuda-10.0" | |
build --action_env TF_CUDNN_VERSION="7" | |
build --action_env TENSORRT_INSTALL_PATH="/usr/lib/x86_64-linux-gnu" | |
build --action_env TF_TENSORRT_VERSION="5.0.0" | |
build --action_env NCCL_INSTALL_PATH="/usr/local/cuda-10.0/targets/x86_64-linux/lib" | |
build --action_env NCCL_HDR_PATH="/usr/local/cuda-10.0/targets/x86_64-linux/lib/../include" | |
build --action_env TF_NCCL_VERSION="2" | |
build --action_env TF_CUDA_COMPUTE_CAPABILITIES="5.0" | |
build --action_env TF_CUDA_CLANG="0" | |
build --action_env GCC_HOST_COMPILER_PATH="/usr/bin/gcc" | |
build --config=cuda | |
test --config=cuda | |
build:opt --copt=-mavx | |
build:opt --copt=-mavx2 | |
build:opt --copt=-mfma | |
build:opt --copt=-mfpmath=both | |
build:opt --host_copt=-march=native | |
build:opt --define with_default_optimizations=true | |
build:v2 --define=tf_api_version=2 |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
This was built with M1000M GPU. With Pascal architecture (GTX 1080..) or later, you can update the environment variable
TF_CUDA_COMPUTE_CAPABILITIES
to6.1
. The latest Turing architecture such as GTX 2080 supports 7.5.https://en.wikipedia.org/wiki/CUDA