diff options
author | Martin Wicke <wicke@google.com> | 2016-09-21 13:16:48 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-09-21 14:31:57 -0700 |
commit | 999b794c137d12d73adbf41dcbe9383a0cd94769 (patch) | |
tree | 90a58d6f358e4dff917912d46cbb1fbf89536ff4 /tensorflow/g3doc/how_tos/adding_an_op/index.md | |
parent | 8e308a8493436fdd7b555837d36459fb232bc7c2 (diff) |
Merge changes from github.
Change: 133874452
Diffstat (limited to 'tensorflow/g3doc/how_tos/adding_an_op/index.md')
-rw-r--r-- | tensorflow/g3doc/how_tos/adding_an_op/index.md | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/g3doc/how_tos/adding_an_op/index.md b/tensorflow/g3doc/how_tos/adding_an_op/index.md index 99787ac95d..a9c29be5e7 100644 --- a/tensorflow/g3doc/how_tos/adding_an_op/index.md +++ b/tensorflow/g3doc/how_tos/adding_an_op/index.md @@ -1000,6 +1000,11 @@ cuda_op_kernel.cu.o -I $TF_INC -fPIC -lcudart `cuda_op_kernel.so` produced above can be loaded as usual in Python, using the `tf.load_op_library` function. +Note that if your CUDA libraries are not installed in `/usr/local/lib64`, +you'll need to specify the path explicitly in the second (g++) command above. +For example, add `-L /usr/local/cuda-8.0/lib64/` if your CUDA is installed in +`/usr/local/cuda-8.0`. + ## Implement the gradient in Python Given a graph of ops, TensorFlow uses automatic differentiation |