使用了V100的算力 但在运行项目时 还是显示使用的CPU
是不是没有独显是不能使用GPU的,但是他提供的不就是GPU吗
使用了V100的算力 但在运行项目时 还是显示使用的CPU
是不是没有独显是不能使用GPU的,但是他提供的不就是GPU吗
上午好☀️☀️☀️️
本答案参考ChatGPT-3.5
在百度飞桨平台使用高级算力的问题中,出现运行项目时仍然使用CPU的情况,可能存在以下原因:
为了解决这个问题,可以进行以下步骤:
修改后的代码:
import paddle.fluid as fluid
with fluid.dygraph.guard():
inputs = fluid.dygraph.to_variable(data) # data为数据集
label = fluid.dygraph.to_variable(labels) # labels为数据标签
model = Net() # Net是自己定义的模型
optimizer = fluid.optimizer.AdamOptimizer(learning_rate=0.001, parameter_list=model.parameters())
with fluid.dygraph.guard(fluid.CUDAPlace(0)): # 选择使用GPU
output = model(inputs)
loss = fluid.layers.cross_entropy(output, label)
avg_loss = fluid.layers.mean(loss)
avg_loss.backward()
optimizer.minimize(avg_loss)
print(avg_loss.numpy()) # 输出训练的损失值