logo
Loading...

有關tensorflow是否有用到gpu的問題? - Cupoy

請問是否只要在jupyter那打入 os.environ["CUDA_VISIBLE_D...

有關tensorflow是否有用到gpu的問題?

2021/02/10 下午 08:45
機器學習共學討論版
鄧書桓
觀看數:150
回答數:2
收藏數:0

請問是否只要在jupyter那打入 os.environ["CUDA_VISIBLE_DEVICES"] = "0" 指令就會切成gpu來跑? 因為本人在輸完上述指令後再跑深度學習的東西,結果cpu使用率還是100%,這樣是有用到嗎? 下面為使用效率![螢幕擷取畫面 2021-02-10 204519.png](http://kwassistfile.cupoy.com/000001778BF5CCDC000001B66375706F795F72656C656173655155455354/1611820870118/large)

回答列表

  • 2021/02/10 下午 09:58
    Jeffrey
    贊同數:0
    不贊同數:0
    留言數:0

    Hello, 你好, 1. os.environ["CUDA_VISIBLE_DEVICES"] = "0" --> 這個指令是說明有一張GPU 顯卡可供使用, 2. 安裝 tensorflow - gpu 版本 3. #使用 30% 的 GPU 記憶體 gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.3)

  • 2021/02/17 下午 05:10
    張維元 (WeiYuan)
    贊同數:0
    不贊同數:0
    留言數:0

    嗨,你好
    你可以直接參考這個文章比較快:https://blog.gtwang.org/programming/tensorflow-keras-specify-gpu-and-memory-tutorial/

    嗨,你好,我是維元,持續在不同的平台發表對 #資料科學、 #網頁開發 或 #軟體職涯 相關的文章。如果對於內文有疑問都歡迎與我們進一步的交流,都可以追蹤我的 Facebook技術部落格 ,也會不定時的舉辦分享活動,一起來玩玩吧 ヽ(●´∀`●)ノ 以下分享一些我近期發表跟資料科學有關的文章,歡迎大家持續追蹤: ■ 資料分析工具那麼多,該怎麼選? 🛠️
    真.資料團隊與分工
    觀察資料的 N 件事 🔖
    資料前處理必須要做的事 - 資料清理與型態調整