[問題]想跑DL(Tensorflow)此GPU條件是否可行?

[kumokochiu (積雨雲)]

作業系統: win10

問題類別: DL(object tracking in video)

使用工具: Python(Anaconda)+tensorflow

問題內容:

我是生物領域的外行人，請各位鞭小力點QQ

我想利用別人已發表在期刊上的opensource code分析小鼠行為，
（Paper: https://www.nature.com/articles/s41593-018-0209-y）
（Github: https://github.com/AlexEMG/DeepLabCut）
使用DL在影片中做animal tracking，安裝步驟包含CUDA和Tensorflow。

實驗室的桌機設備：Win10, GPU:NVIDIA GTX1060 3GB
但我爬文看到人家說跑Tensorflow只有GTX1060 6GB版本可能跑得動，
想請教有經驗的人
我這個code在此GPU上有可能跑得動嗎？（要分析的影片長度大約15分鐘）

（無知如我很怕亂跑程式把實驗室電腦搞壞QQQ）

如果有觀念不對的地方麻煩不吝指正
感激不盡！

—
補上實驗室桌機其他資訊
https://i.imgur.com/GBH0CSi.jpg
--
※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 120.126.102.57
※ 文章網址: https://www.ptt.cc/bbs/DataScience/M.1548232923.A.652.html
※ 編輯: kumokochiu (27.52.39.144), 01/23/2019 16:51:35
推 sean50301: 1050也有人在跑 不用擔心 反而很好奇是什麼文章寫的一
→ sean50301: 定要1060 6g
也不是說一定 應該說我看分析文提到1060都只推6g
推 school4303: 最多跑不起來 電腦不會有事啦
推 sean50301: 反而裝環境很容易把作業系統搞爆XD 雖然現在tf已經算好
→ sean50301: 裝了
本人那天愚蠢地跑去把driver update到最新的版本 結果安裝完電腦就黑屏
後來才知driver求穩不求新 復原舊版本是沒事了 但我命差不多也嚇掉半條...
推 Kazimir: 沒仔細看 不過假設他要把resnet都讀進去那我覺得能跑的
→ Kazimir: 機會渺茫 雖然要看你的影片解析度 不過3gb應該很難吧
感謝回答！！
※ 編輯: kumokochiu (120.126.102.57), 01/23/2019 17:55:36
※ 編輯: kumokochiu (120.126.102.57), 01/23/2019 17:59:51
推 harry0731: 找找看有沒有pretrained model 有的話只需要做inferen
→ harry0731: ce應該有機會 大不了用cpu跑
→ f496328mm: 3G ram 太小拉
推 ZongXiu: 用CPU先跑看看吧，模型不大時使用GPU不一定會比較快
推 zxp9505007: Cpu一定可行 速度看你能不能接受
推 alen84204: 同問 32g ram I5 550TI 跑的動嗎QQ
推 Morphee: Win10 何苦
推 CarbonK: 只有3G你自己建模型要先算一下，超過的話就不行
→ CarbonK: 當然你對IO有信心也能分一部分到CPU去啦
推 f496328mm: 跑DL，要看GPU的ram，不然你會慢到哭死，至少 1070
→ Mchord: gpu的ram影響的不是速度吧…時脈跟cuda core數才是
→ Mchord: gpu ram主要影響可以用的model大小，包含input尺寸以及bat
→ Mchord: ch size。
→ Mchord: 跑不動頂多跳OOM就是了，不會壞掉。
推 f496328mm: ram影響batch，batch影響速度
推 zxp9505007: Batch大 收斂快 當然影響速度
→ Mchord: 原po不是在問跑不跑得動嗎？訓練速度也不是batch大就一定
→ Mchord: 最快，所需update次數少跟整體訓練時間是不一樣的意義。
推 CarbonK: 所以才說要自己估一下啊
→ CarbonK: 反正 global_variables 能拿到所有參數的尺寸
→ CarbonK: 算算看有幾個乘四大概就知道要幾 B 了
→ jackwang01: 會推6G主要是cp值很棒，往上1080價錢比較不親民，往下
→ jackwang01: 1060 3G沒省多少又有可能train不動
→ RumiManiac: 3G 和 6G 天跟地的差別啊
推 sma1033: 再往上的1070也不錯啦，記憶體跟cuda都有增加
推 sma1033: 1063真是慘的不忍直視