oarriaga/paz

Minimal Hand: demo.py , CUBLAS_STATUS_ALLOC_FAILED

malwaru opened this issue · 1 comments

malwaru commented

Could not run the script minimal_hand/examples/demo.py because of the error "CUBLAS_STATUS_ALLOC_FAILED"

malwaru commented

Fixed the issue by adding the following lines of code add the top of the file

import tensorflow as tf
gpus = tf.config.experimental.list_physical_devices('GPU')
tf.config.experimental.set_memory_growth(gpus[0], True)