如何解决运行时错误:CUDA 错误:python 3.7 未指定启动失败
训练过程开始。为了避免cuda内存不足错误,我设置了1为batch size。
但是经过几个时期后,有时甚至大部分时间,我都会收到 RuntimeError: CUDA error: unspecified launch failure
错误。这非常令人沮丧。
每次出现错误时,输入大小都会发生变化,因此不会发生相同的输入。
train.py:
for i,(inputs,target,_) in enumerate(train_loader):
print(torch.cuda.is_available())
input_var = [input.cuda() for input in inputs]
target_var = target.cuda()
output = model(input_var)
loss = criterion(output,target_var)
losses.update(loss.item(),1)
# compute accuracy
prec1,prec5 = accuracy(output.data.cpu(),topk=(1,5))
top1.update(prec1[0].item(),1)
top5.update(prec5[0].item(),1)
# zero the parameter gradients
optimizer.zero_grad()
# compute gradient
loss.backward()
optimizer.step()
.....
输出:
True
136
Traceback (most recent call last):
File "....\train.py",line 273,in <module>
train(train_loader,model,criterion,optimizer,epoch)
File ".....\train.py",line 75,in train
input_var = [input.cuda() for input in inputs]
File "......\train.py",in <listcomp>
input_var = [input.cuda() for input in inputs]
RuntimeError: CUDA error: unspecified launch failure
你知道我该如何解决这个错误吗? 谢谢。
- Windows 10
- NVIDIA GeForce GTX 1060
- 火炬 1.6
- CUDA 10.1
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。