如何调用GPU进行训练?
收藏
】
我训练的时候一个batch 18秒左右 比我电脑上跑的还慢 想问一下怎么调用tesla的卡进行训练? 我的tf是gpu版的
0
收藏
请登录后评论
运行项目的时候选择tesla卡即可
AI Studio 平台上训练
在运行项目时,选择GPU配置就可以,不需要代码上的操作
代码指定使用cuda的device 平台上用GPU
同样的问题 环境进入的时候选择的是GPU环境 代码上TF 海需要调用吗?
aistudio 不支持tf
运行项目的时候,选择GPU那个算力就行