#问题
在跑深度学习代码时,本来应该是在GPU上跑,但是实际是在CPU上跑,google了一下,发现模型和输入以下述格式调用GPU:
model.cuda()
这个是model调用GPU
inputs = Variable(inputs.cuda())
这个是输入调用GPU。
#问题
在跑深度学习代码时,本来应该是在GPU上跑,但是实际是在CPU上跑,google了一下,发现模型和输入以下述格式调用GPU:
model.cuda()
这个是model调用GPU
inputs = Variable(inputs.cuda())
这个是输入调用GPU。