Skip to content

Instantly share code, notes, and snippets.

@neofob
Created November 1, 2018 03:42
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save neofob/103e6a45d1346915405c9e1f6dab3886 to your computer and use it in GitHub Desktop.
Save neofob/103e6a45d1346915405c9e1f6dab3886 to your computer and use it in GitHub Desktop.
tensorflow stuff
# 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
# with bazel 0.19, add tools/bazel.rc as below
import /tmp/tensorflow/tools/bazel.rc
import /tmp/tensorflow/.tf_configure.bazelrc
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
@neofob
Copy link
Author

neofob commented Nov 1, 2018

This was built with M1000M GPU. With Pascal architecture (GTX 1080..) or later, you can update the environment variable TF_CUDA_COMPUTE_CAPABILITIES to 6.1. The latest Turing architecture such as GTX 2080 supports 7.5.

https://en.wikipedia.org/wiki/CUDA

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment