There seems to be a bug in the current CUBLAS library implementation of cublas<t>getrfBatched for matrices of dimension (n) such that 3<=n<=16, when there is a “zero pivot” as you say. A possible workaround is to “identity-extend” your A matrix to be inverted, when n<17, to a size of 17×17 (using matlab nomenclature): LU = … Read more
For TensorFlow 2.2 none of the other answers worked when the CUBLAS_STATUS_ALLOC_FAILED problem was encountered. Found a solution on https://www.tensorflow.org/guide/gpu: import tensorflow as tf gpus = tf.config.experimental.list_physical_devices(‘GPU’) if gpus: try: # Currently, memory growth needs to be the same across GPUs for gpu in gpus: tf.config.experimental.set_memory_growth(gpu, True) logical_gpus = tf.config.experimental.list_logical_devices(‘GPU’) print(len(gpus), “Physical GPUs,”, len(logical_gpus), “Logical … Read more