CUBLAS_STATUS_ALLOC_FAILED 解決辦法

訓練facenet中的train_softmax.py時，出現瞭如下錯誤session failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED 這個是因爲GPU的配置不對形成的。spa 解決辦法：code config = tf.ConfigProto() config.gpu_options.allow_growth = True s

>>阅读原文<<