GPU使用问题
收藏
在AI Studio的GPU环境下运行具有大量矩阵乘法的python代码时，能够像Google Colab那样实现加速吗？在google colab中，在GPU环境下直接运行python代码的速度比CPU环境快很多。
若想在AI studio中实现GPU加速，是不是一定需要用paddle将代码重写？代码中的变量非常之多。
希望哪位好心人能给个解答。万分感谢。
0
收藏
请登录后评论
实践出真知, 你把那个代码copy过来试试呗
应该有类似语法，比如变量a，用a.cuda()转换到gpu设备上运行（这是pytorch的语法，paddle的我没涉及这方面），只要你的矩阵是variable或tensor类型，在需要的部分添(比如创建矩阵的地方)加上.cuda()就好了，不用非得全部重写