Torch 带gpu的火炬列车始终正确_Torch

Torch 带gpu的火炬列车始终正确

Torch 带gpu的火炬列车始终正确,torch,Torch,我是火炬新手，我已在Ubuntu14.04上安装了torch7，并安装了cuda8.0和cudnn5.1。我用CPU尝试了mnist和cifar10，效果很好。我使用的代码，train on digital classifier和train on cifar，都处于启用状态。在使用GPU进行培训之前，我添加了以下代码 cutorch.setDevice(1) model = model:cuda() criterion = criterion:cuda() trainData.data = t

我是火炬新手，我已在

Ubuntu14.04

上安装了

torch7

，并安装了

cuda8.0

和

cudnn5.1

。我用CPU尝试了

mnist

和

cifar10

，效果很好。我使用的代码，

train on digital classifier

和

train on cifar

，都处于启用状态。在使用GPU进行培训之前，我添加了以下代码

cutorch.setDevice(1)
model = model:cuda()
criterion = criterion:cuda()
trainData.data = trainData.data:cuda()
trainData.labels = trainData.labels:cuda()
testData.data = testData.data:cuda()
testData.labels = testData.labels:cuda()

它可以运行，但正确的方法在每个时代都是一样的。有人能告诉我哪里错了吗