logo
Loading...

要如何確保目前執行Keras是使用電腦的GPU在跑? - Cupoy

各位專家好: 想請問若是今天在跑深度學習的Keras的程式碼 要如何才能知道目前是用電腦的GPU在跑...

要如何確保目前執行Keras是使用電腦的GPU在跑?

2021/08/25 上午 11:09
Keras 安裝與介紹
Yaoga
觀看數:625
回答數:1
收藏數:0

各位專家好: 想請問若是今天在跑深度學習的Keras的程式碼 要如何才能知道目前是用電腦的GPU在跑還是CPU? 另外若是CPU在跑的話,要如何轉換成GPU 用GPU跑是否速度成效上較為穩定? 另外額外想問一個問題,若是筆電採用Anaconda來跑深度學習, 筆電的GPU規格至少要到什麼樣的等級跑的速度才會必較快? 感謝~

回答列表

  • 2021/08/25 上午 11:32
    王健安
    贊同數:0
    不贊同數:0
    留言數:2

    Yaoga您好, 確認深度學習訓練是否有使用GPU, 最直接的方法是透過「工作管理員」→「效能」中的GPU是否有運行, 另外tensorflow也有一些偵測是否有GPU的方法,例如:tensorflow.test.is_gpu_available, 至於如何轉換成GPU 可以使用以下程式碼: ```python import os os.environ["CUDA_VISIBLE_DEVICES"] = "0" ``` 關於tensorflow與GPU的使用, 先前助教時間有更完整的介紹, 歡迎參閱。 https://www.cupoy.com/post/0000017AD7548998000000066375706F795F72656C656173654B5741535354434C55424E455753