返回信息流[ema1]本人小白一枚…想在自己电脑上装一个GPU版本的tensorflow,请教一下Docker装镜像/windows上配tensorflow/装Linux虚拟机再配tensorflow哪种会更好些…
这是一条镜像帖。来源:北邮人论坛 / ml-dm / #28231同步于 2018/2/8
该镜像源已超过 30 天没有更新,可能在源站已被删除。
ML_DM机器人发帖
求教tensorflow安装问题
czl9794
2018/2/8镜像同步11 回复
订阅后,新回复会通过你的通知中心匿名送达。
9 条回复
windows上你还是别装GPU版本了吧~你也装不上!
第一点:如果你想简单的跑跑少量数据集,测试基本的模型,我觉得cpu版本就够了,把数据集缩小一些来试试,千万不要整个上G的数据,你的小电脑会被你玩坏的。真要跑GPU,去亚马逊官方提供的租一个还是可以的,好像也不是特别的贵,租一个的话我觉得还是物有所值的,不过需要注意的是不用的时候千万记得手动关掉,因为GPU即使训练完了它也不是停的。亚马逊上是按照你使用GPU的时间来计价的,所以不用的时候关掉就好了,用的话先在自己的小电脑上把模型准确性之类的测试好啦,再拿上去跑数据会比较好。
第二点:个人觉得只是简单学习模型的话,我强烈推荐通过anaconda在windows下安装tensorflow-cpu的版本,通过anaconda,你可以自己建很多虚拟的那种环境,每种环境下你还能安装不同的版本(tensorflow不同版本真的很重要),具体anaconda怎么操作查看官方文档或者很多博客有写。个人建议follow官方的安装教程和create envs的教程。出问题了也比较好找答案。
只选ubuntu
如果你电脑只自己用的话用不用docker我感觉都没有区别.从头从显卡驱动, cuda, cudnn这几个基础设施装好 然后用pip3(我只用python3) 装tensorflow-gpu就好.
基础设施和tensorflow版本之间要相互符合.比如tf1.4对应cuda8 tf1.5对应cuda9这样的.
坑肯定是要趟一遍的,最好按照网上别人的经验严格按照版本安装好.
windows好像只支持支付宝python3的某个版本吧,linux相对来说比较容易一点,反正基本上版本对了直接pip安装就行。不过虚拟机的问题是可能跑比较复杂的模型跑不起来
发自「贵邮」