AIStudio中如何使用GPU进行PARL训练

本篇文章给大家分享的是有关AI Studio中如何使用GPU进行PARL训练,小编觉得挺实用的,因此分享给大家学习,希望大家阅读完这篇文章后可以有所收获,话不多说,跟着小编一起来看看吧。

宜丰ssl适用于网站、小程序/APP、API接口等需要进行数据传输应用场景,ssl证书未来市场广阔!成为创新互联的ssl证书销售渠道,可以享受市场价格4-6折优惠!如果有意向欢迎电话联系或者加微信:028-86922220(备注:SSL证书合作)期待与您的合作!

1.准备篇

PARL是基于paddlepaddle框架的,由于paddlepaddle针对cuda9和cuda10有不同的安装包,所以首先要检查一下AI Studio中的cuda版本。

在AI Studio中打开任意一个项目,在终端中检查一下cuda版本

AI Studio中如何使用GPU进行PARL训练

发现Cuda是9.2版本的. 再检查一下cuDNN版本

AI Studio中如何使用GPU进行PARL训练

发现cuDNN的版本是7.3.1.

根据PP官方安装文档的说明,( https://www.paddlepaddle.org.cn/install/quick)

AI Studio中如何使用GPU进行PARL训练

得出结论, 我们应该使用CUDA 9系列的pp gpu版本, 即使用以下命令来安装

python3 -m pip install paddlepaddle-gpu==1.6.3.post97 -i https://mirror.baidu.com/pypi/simple

注意,要选择post97版本, 不要使用post107

2.操作篇

进入notebook, 先安装一下GPU版本的paddle,例如:

AI Studio中如何使用GPU进行PARL训练

注意上图红框是关键步骤.

后面就正常写代码就行了. 但是,如果你仍然怀疑GPU到底起作用没有的话,可以修改Agent,打印运行环境看看, 来增强一下信心.例如:

AI Studio中如何使用GPU进行PARL训练

注意上面红框中我加的代码, 在运行时可以在日志中看到实际环境,如下:

AI Studio中如何使用GPU进行PARL训练

到此, 可以确定真的是GPU在跑啦.

以上就是AI Studio中如何使用GPU进行PARL训练,小编相信有部分知识点可能是我们日常工作会见到或用到的。希望你能通过这篇文章学到更多知识。更多详情敬请关注创新互联行业资讯频道。


新闻名称:AIStudio中如何使用GPU进行PARL训练
网页地址:http://scyanting.com/article/igieoj.html