請問這些CODE有辦法使用GPU加速嗎?
2020/09/08 下午 02:45
《用 Python 打造你的 AI 股票交易引擎》業界專家實戰教學
王承漢
觀看數：7
回答數：1
收藏數：0
gpu
這裡面各個程式不知道有沒辦法用GPU加速？
當初我以為他是吃tensorflow-gpu結果跑出來只是用CPU在跑...(原諒我是新手
回答列表
-
2020/09/14 下午 04:24陳明佑 (Ming You Chen)贊同數：0不贊同數：0留言數：0
目前版本的程式中，只有GRU模型的部分是用GPU加速的
也就是Part2訓練模型的部分當中的2d模型
訓練到那邊的時候，您就會看到GPU的使用率提高
至於Part1的爬蟲收集資料，以及Part3預測的大部分程式
(Part3用2D模型預測的時候，才會用到GPU)
都只會用到CPU沒錯