Tensorflow GPU 運算出現 failed to create cublas handle 訊息
用GPU RUN PYTHON時,出現錯誤訊息
failed to create cublas handle: cublas_status_alloc_failed
解決方法
1. 指定 GPU 顯示卡
import os
# 使用第一張與第三張 GPU 卡
os.environ["CUDA_VISIBLE_DEVICES"] = "0,2"
2. 指定 GPU 記憶體的佔用量
import tensorflow as tf
# 只使用 30% 的 GPU 記憶體
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.3)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
# 設定 Keras 使用的 TensorFlow Session
tf.keras.backend.set_session(sess)
# 使用 Keras 建立模型
# ..
3.自動增長 GPU 記憶體用量
import tensorflow as tf
# 自動增長 GPU 記憶體用量
gpu_options = tf.GPUOptions(allow_growth=True)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
# 設定 Keras 使用的 Session
tf.keras.backend.set_session(sess)
# 使用 Keras 建立模型
# ..
4. 如果還是不行
換CUDA跟CUDNN版本
文章出處: NetYea網頁設計
頁:
[1]