Tensorflow GPU 運算出現 failed to create cublas handle 訊息

woff

用GPU RUN PYTHON時，
出現錯誤訊息

1. failed to create cublas handle: cublas_status_alloc_failed

複製代碼

解決方法
1. 指定 GPU 顯示卡

1. import os
   
2. # 使用第一張與第三張 GPU 卡
   
3. os.environ["CUDA_VISIBLE_DEVICES"] = "0,2"
   

複製代碼

2. 指定 GPU 記憶體的佔用量

1. import tensorflow as tf
   
2. # 只使用 30% 的 GPU 記憶體
   
3. gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.3)
   
4. sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
   
5. # 設定 Keras 使用的 TensorFlow Session
   
6. tf.keras.backend.set_session(sess)
   
7. # 使用 Keras 建立模型
   
8. # ..

複製代碼

3.自動增長 GPU 記憶體用量

1. import tensorflow as tf
   
2. # 自動增長 GPU 記憶體用量
   
3. gpu_options = tf.GPUOptions(allow_growth=True)
   
4. sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
   
5. # 設定 Keras 使用的 Session
   
6. tf.keras.backend.set_session(sess)
   
7. # 使用 Keras 建立模型
   
8. # ..

複製代碼

4. 如果還是不行
換CUDA跟CUDNN版本

文章出處： NetYea網頁設計