首页 PaddleRec 帖子详情
关于GPU问题
收藏
快速回复
PaddleRec 问答学习资料 609 1
关于GPU问题
收藏
快速回复
PaddleRec 问答学习资料 609 1

就是

paddle.device.set_device("gpu:0")

之后,我看文档说开启gpu使用,但是实际训练中还是用cpu,请问这是为啥呢

引用的库是

import paddle
from paddlenlp.datasets import load_dataset
import paddle.nn.functional as F
import paddle.nn as nn
import paddlenlp as ppnlp
from paddlenlp.transformers import BertPretrainedModel, BertTokenizer,BertForSequenceClassification
from paddlenlp.transformers import LinearDecayWithWarmup

 

0
收藏
回复
全部评论(1)
时间顺序
FutureSI
#2 回复于2022-03

paddle的gpu版本,以及cuda,cudnn都装好自动就会启用gpu

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户