使用如下源代码的时候发生了如下错误该如何修改?当前安装的平台是kaggle源代码:!pip install paddlepaddle -i httpsmirrorbaiducompypisimple# 数据科学包import random # 随机切分数据集import numpy as np # 常用数据科学包import
根据错误提示,PaddlePaddle没有编译CUDA,因此不能使用GPU。需要将设备设置为CPU,即将以下代码:
paddle.set_device('gpu:0')
修改为:
paddle.set_device('cpu')
另外,还需要修改模型最后一层的输出大小,将原来的1000改为12,即:
model.fc = paddle.nn.Linear(in_features=2048, out_features=12)
原文地址: https://www.cveoy.top/t/topic/hePM 著作权归作者所有。请勿转载和采集!