tf.GPUOptions.RegisterExtension(extension_handle)

tf.GPUOptions.RegisterExtension(extension_handle)