复盘《pytorch实战》--bug+遗漏知识点（五）

ch5.深度学习之计算机视觉

1.ImageFolder（）

用来读图片时，如果一级文件有几个文件夹，文件夹里面是图片，那么路径是一级文件夹，二级文件夹会变成对应的类别。

2.IndexError: invalid index of a 0-dim tensor

原因：train_loss += loss.data[0] 是pytorch0.3.1版本代码,在0.4-0.5版本的pytorch会出现警告,不会报错,但是0.5版本以上的pytorch就会报错。

将原来的语句修改为：

running_loss += F.nll_loss(output,target,size_average=False).item()

3.cuda启动超时

CUDA error: the launch timed out and was terminated

1.打开注册表。

win+r 调出cmd，输入regedit进入注册表。

2.“HKEY_LOCAL_MACHINE\System\CurrentControlSet\Control\GraphicsDrivers“下新建文件，文件名称为”TdrDelay“，文件类型为DWORD。双击文件，修改TDR值。

3.重启电脑

4.解决上述问题之后又出现的问题