无效的设备序号，CUDA/TORCH

Question

我在Ubuntu16.04中运行脚本时遇到这个错误。请原谅我，我是Python新手， 我已经检查了互联网上已经可用的选项，但我无法修复它。在

 RuntimeError: cuda runtime error (10) : invalid device ordinal at torch/csrc/cuda/Module.cpp:32

我当前正在运行此文件。在

^{pr2}$

我怀疑这个文件和弹出的错误有关

Title = "TOML Example"

[general]
usecudnn = true
usecudnnbenchmark = true
gpuid = 0
loadpretrainedmodel = true
pretrainedmodelpath = "trainedmodel.pt"
savemodel = true
modelsavepath = "savedmodel.pt"

[input]
batchsize = 18
numworkers = 18
shuffle = true

[model]
type = "LSTM"
inputdim = 256 
hiddendim = 256
numclasses = 500
numlstms = 2

[training]
train = true
epochs = 15
startepoch = 10
statsfrequency = 1000
dataset = "/udisk/pszts-ssd/AV-ASR-data/BBC_Oxford/lipread_mp4"
learningrate = 0.003
momentum = 0.9
weightdecay = 0.0001

[validation]
validate = true
dataset = "/udisk/pszts-ssd/AV-ASR-data/BBC_Oxford/lipread_mp4"
saveaccuracy = true
accuracyfilelocation = "accuracy.txt"

错误主要在gpuid行中，因为我终于到达了。在

Answer 1

如果预先训练的模型在不同数量的Cuda设备上训练，则可能会出现该错误。例如，在训练模型时，您使用了3个Cuda设备，而现在您正在一个只有一个Cuda设备的设备上加载相同的训练模型。在