训练facenet中的train_softmax.py时,出现了以下错误
failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED
这个是由于GPU的配置不对造成的。
解决办法:
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config, ...)
训练facenet中的train_softmax.py时,出现了以下错误
failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED
这个是由于GPU的配置不对造成的。
解决办法:
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config, ...)