AI Studio下如何跑tensorflow

您所在的位置:网站首页 飞桨gpu AI Studio下如何跑tensorflow

AI Studio下如何跑tensorflow

2023-09-08 07:05| 来源: 网络整理| 查看: 265

AI Studio是百度开发的基于其深度学习框架paddlepaddle(飞桨)的AI一站式平台,通过为大家提供免费的顶级GPU(Tesla v100)来推广飞桨.

AI Studio可以说是目前国内最好的提供GPU的平台,我自己为了寻找免费的GPU,尝试过Azure,AWS等平台,饱受外文网站折磨,看到AI Studio的时候不禁感叹百度终于tm干了件人事。 但是AI Studio Notebook中预装的是飞桨框架,不支持TF,难道我为了用这个GPU还必须要把代码改成Paddlepaddle的吗?也太麻烦了。但是技术说只要操作正确,TF也是可以跑起来的,于是我就开始了作死的尝试。。。 直接在cell里

!pip install tensorflow

这样装只是临时的,重启就没了,持久化安装第三方包的正确姿势是

!mkdir /home/aistudio/external-libraries !pip install yourpackage -t /home/aistudio/external-libraries

yourpage就是你想安装的包,然后运行程序时加上这句话

import sys sys.path.append(’/home/aistudio/external-libraries’) 这样装上TF就能用了?这么简单我也不会写教程了,装上TF后你运行它一定会报错:

TypeError: init() got an unexpected keyword argument ‘serialized_options’

这是为啥呢,其实是protobuf这个包的版本问题,服务器预装的protobuf版本是3.1.0(好像),然后你可以用: !pip install -U protobuf 将服务器中的这个包升级到3.9.0。运气好的话这个时候你的TF就能用了,import tensorflow不会报错,但是多次重启项目之后你会发现一件很迷的事情!就是明明刚刚不报错,一会儿它又报错了!!!十分鬼畜,我研究了很久,做了很多实验得出一个暂时能用的结论和方法,就是能不能跑tf和他的服务器有关,如果你运行项目之后不能跑就尝试重新安装一下TF和protobuf吧,还不行就重启项目,几次之后就可以了。 然后就可以薅百度的羊毛了,用着他的GPU跑tensorflow,是真的快,十几分钟跑出了我pcCPU两晚上的成果。不过还是建议大家不要一直白嫖,AI Studio真的是个很好的平台,支持国产呀,学学paddlepaddle!

最后说一下Azure,微软这东西感觉就不适合个人用,很多是面向企业的,他的学生免费还没对大陆高校的edu.cn邮箱开放,也就是说中国学生申请不到学生试用的,就算你申请到了GitHub的教育包,给了验证码,拿到Azure那边去还是不能用。 我折腾了好久还是用自己信用卡注册了,但是说实话真的不太会,创建了虚拟机想跑程序还是需要点技术,我尝试了一下还是放弃了,如果哪位大神知道怎么白嫖Aure的GPU麻烦告知小弟一下大概流程,谢谢。



【本文地址】


今日新闻


推荐新闻


CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3