作者kumokochiu (积雨云)
看板DataScience
标题[问题]想跑DL(Tensorflow)此GPU条件是否可行?
时间Wed Jan 23 16:41:59 2019
作业系统: 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://webptt.com/cn.aspx?n=bbs/DataScience/M.1548232923.A.652.html
※ 编辑: kumokochiu (27.52.39.144), 01/23/2019 16:51:35
1F：推 sean50301: 1050也有人在跑 不用担心 反而很好奇是什麽文章写的一 01/23 16:51
2F：→ sean50301: 定要1060 6g 01/23 16:51
也不是说一定 应该说我看分析文提到1060都只推6g
3F：推 school4303: 最多跑不起来 电脑不会有事啦 01/23 17:01
4F：推 sean50301: 反而装环境很容易把作业系统搞爆XD 虽然现在tf已经算好 01/23 17:27
5F：→ sean50301: 装了 01/23 17:27
本人那天愚蠢地跑去把driver update到最新的版本 结果安装完电脑就黑屏
後来才知driver求稳不求新 复原旧版本是没事了 但我命差不多也吓掉半条...
6F：推 Kazimir: 没仔细看 不过假设他要把resnet都读进去那我觉得能跑的 01/23 17:40
7F：→ Kazimir: 机会渺茫 虽然要看你的影片解析度 不过3gb应该很难吧 01/23 17:41
感谢回答！！
※ 编辑: kumokochiu (120.126.102.57), 01/23/2019 17:55:36
※ 编辑: kumokochiu (120.126.102.57), 01/23/2019 17:59:51
8F：推 harry0731: 找找看有没有pretrained model 有的话只需要做inferen 01/23 19:25
9F：→ harry0731: ce应该有机会 大不了用cpu跑 01/23 19:25
10F：→ f496328mm: 3G ram 太小拉 01/23 19:46
11F：推 ZongXiu: 用CPU先跑看看吧，模型不大时使用GPU不一定会比较快 01/23 21:05
12F：推 zxp9505007: Cpu一定可行 速度看你能不能接受 01/23 21:29
13F：推 alen84204: 同问 32g ram I5 550TI 跑的动吗QQ 01/23 22:58
14F：推 Morphee: Win10 何苦 01/23 23:22
15F：推 CarbonK: 只有3G你自己建模型要先算一下，超过的话就不行 01/24 00:09
16F：→ CarbonK: 当然你对IO有信心也能分一部分到CPU去啦 01/24 00:09
17F：推 f496328mm: 跑DL，要看GPU的ram，不然你会慢到哭死，至少 1070 01/24 01:21
18F：→ Mchord: gpu的ram影响的不是速度吧…时脉跟cuda core数才是 01/24 07:26
19F：→ Mchord: gpu ram主要影响可以用的model大小，包含input尺寸以及bat 01/24 07:29
20F：→ Mchord: ch size。 01/24 07:29
21F：→ Mchord: 跑不动顶多跳OOM就是了，不会坏掉。 01/24 07:32
22F：推 f496328mm: ram影响batch，batch影响速度 01/24 11:20
23F：推 zxp9505007: Batch大 收敛快 当然影响速度 01/24 23:04
24F：→ Mchord: 原po不是在问跑不跑得动吗？训练速度也不是batch大就一定 01/24 23:58
25F：→ Mchord: 最快，所需update次数少跟整体训练时间是不一样的意义。 01/24 23:58
26F：推 CarbonK: 所以才说要自己估一下啊 01/25 00:07
27F：→ CarbonK: 反正 global_variables 能拿到所有参数的尺寸 01/25 00:07
28F：→ CarbonK: 算算看有几个乘四大概就知道要几 B 了 01/25 00:07
29F：→ jackwang01: 会推6G主要是cp值很棒，往上1080价钱比较不亲民，往下 01/25 03:44
30F：→ jackwang01: 1060 3G没省多少又有可能train不动 01/25 03:44
31F：→ RumiManiac: 3G 和 6G 天跟地的差别啊 01/26 08:30
32F：推 sma1033: 再往上的1070也不错啦，记忆体跟cuda都有增加 01/28 12:08
33F：推 sma1033: 1063真是惨的不忍直视 01/28 16:34