本篇文章给大家分享的是有关AI Studio中如何使用GPU进行PARL训练，小编觉得挺实用的，因此分享给大家学习，希望大家阅读完这篇文章后可以有所收获，话不多说，跟着小编一起来看看吧。
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训练，小编相信有部分知识点可能是我们日常工作会见到或用到的。希望你能通过这篇文章学到更多知识。更多详情敬请关注亿速云行业资讯频道。
免责声明：本站发布的内容（图片、视频和文字）以原创、转载和分享为主，文章观点不代表本网站立场，如果涉及侵权请联系站长邮箱：is@yisu.com进行举报，并提供相关证据，一经查实，将立刻删除涉嫌侵权内容。
