24h購物| | PChome| 登入
2023-10-07 10:08:50| 人氣3| 回應0 | 上一篇 | 下一篇
推薦 0 收藏 0 轉貼0 訂閱站台

Tensorflow GPU 運算出現 failed to

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

  1. failed to create cublas handle: cublas_status_alloc_failed
複製代碼

1.png


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

  1. import os
  2. # 使用第一張與第三張 GPU 卡
  3. os.environ["CUDA_VISIBLE_DEVICES"] = "0,2"
  4.  
網站架設 複製代碼



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網頁設計

我要檢舉
台長： jamieyw2

您可能對以下文章有興趣

人氣(3) | 回應(0)| 推薦 (0)| 收藏 (0)| 轉寄
全站分類: 健康樂活(醫學、養生、減重) | 個人分類: NetYea |
此分類下一篇:網站架設 CPanel 服務器用transfer Tool對
此分類上一篇:Yolov8 Models 測試

我要回應
否 是 (若未登入"個人新聞台帳號"則看不到回覆唷!)
* 請輸入識別碼：
請輸入圖片中算式的結果(可能為0) 
(有*為必填)
TOP
詳全文