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

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

1.准备篇

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

2.操作篇

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

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

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

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

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

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