How to build CUDA JIT caches for all available kernels in TensorFlow programmatically?
<p>I encountered the "first-run slow-down" problem with GTX 1080 cards and nvidia-docker as discussed in <a href="https://stackoverflow.com/questions/36842169/">this question</a>.</p>

<p>I'm using the TensorFlow build from <a href="https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.11.0rc2-cp35-cp35m-linux_x86_64.whl" rel="nofollow noreferrer">its official pip package</a> and a custom docker image based on nvidia-docker's Ubuntu 16.04 base image.</p>

<p>How do I make TensorFlow to load (and build JIT caches) all registered CUDA kernels programmatically in a Dockerfile? (rather than manually building TensorFlow using <code>TF_CUDA_COMPUTE_CAPABILITIES</code> environment variable)</p>