20/9/14:跑代码
1、python模块以及导入出现ImportError: No module named 'xxx’问题
解决:
1、在同目录模块前面加.
或者加.包名
2、最后发现是用的Python3的解释器,换用python2的
2、runtimeError: cuda runtime error (38): no CUDA-capable device is detected at XXXX
解决:可能是torch和torchvision与cuda版本的问题。但我这里是因为没有指定gpuid,若只有一块gpu则gpuid为0
3、相对路径加载文件表示找不到文件
这里没有解决,用的绝对路径
4、IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python
loss.data[0] 是pytorch0.3.1版本代码,在0.4-0.5版本的pytorch会出现警告,不会报错,但是0.5版本以上的pytorch就会报错,总的来说是版本更新问题.
解决:.data[0]
改为.item()
原语句:running_loss+=loss.data[0]
修改后:running_loss+=loss.item()