关于GPU问题
收藏
就是
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
收藏
请登录后评论
paddle的gpu版本,以及cuda,cudnn都装好自动就会启用gpu