Hello! I am trying to find which library the GPU kernels are compiled into as its not in libtensorflow_framework.so.2
Also if I implemented a custom op, can I used NVCC to compile it for the GPU?
Hi @feeloats
Welcome to the TensorFlow Forum!
Generally, GPU kernels libraries resides in a system library directory where all other TensorFlow libraries installs. And yes, you can use the NVIDIA Compiler (NVCC) to compile custom TensorFlow operations for GPU execution. Please refer to the TensorFlow doc on how to Create an op for more details in this. Thank you