十年网站开发经验 + 多家企业客户 + 靠谱的建站团队
量身定制 + 运营维护+专业推广+无忧售后,网站问题一站解决
本篇文章给大家分享的是有关AI Studio中如何使用GPU进行PARL训练,小编觉得挺实用的,因此分享给大家学习,希望大家阅读完这篇文章后可以有所收获,话不多说,跟着小编一起来看看吧。
宜丰ssl适用于网站、小程序/APP、API接口等需要进行数据传输应用场景,ssl证书未来市场广阔!成为创新互联的ssl证书销售渠道,可以享受市场价格4-6折优惠!如果有意向欢迎电话联系或者加微信:028-86922220(备注:SSL证书合作)期待与您的合作!
PARL是基于paddlepaddle框架的,由于paddlepaddle针对cuda9和cuda10有不同的安装包,所以首先要检查一下AI Studio中的cuda版本。
在AI Studio中打开任意一个项目,在终端中检查一下cuda版本
发现Cuda是9.2版本的. 再检查一下cuDNN版本
发现cuDNN的版本是7.3.1.
根据PP官方安装文档的说明,( https://www.paddlepaddle.org.cn/install/quick)
得出结论, 我们应该使用CUDA 9系列的pp gpu版本, 即使用以下命令来安装
python3 -m pip install paddlepaddle-gpu==1.6.3.post97 -i https://mirror.baidu.com/pypi/simple
注意,要选择post97版本, 不要使用post107
进入notebook, 先安装一下GPU版本的paddle,例如:
注意上图红框是关键步骤.
后面就正常写代码就行了. 但是,如果你仍然怀疑GPU到底起作用没有的话,可以修改Agent,打印运行环境看看, 来增强一下信心.例如:
注意上面红框中我加的代码, 在运行时可以在日志中看到实际环境,如下:
到此, 可以确定真的是GPU在跑啦.
以上就是AI Studio中如何使用GPU进行PARL训练,小编相信有部分知识点可能是我们日常工作会见到或用到的。希望你能通过这篇文章学到更多知识。更多详情敬请关注创新互联行业资讯频道。