报错Usage: THEANO_FLAGS=“device=gpu0”
一定要Debug调试看看哪个位置报错(之前一直以为是Theano配置问题走了很多弯路)
if len(sys.argv) == 1:
print(__doc__)
sys.exit(-1)
# nb_residual_unit = 2 # number of residual units
else:
nb_residual_unit = int(sys.argv[1]) # number of residual units
报错原因:sys.argv[]用法问题,直接输出到print(doc),输出的就是
THEANO_FLAGS="device=gpu0"
关于sys.arg[]用法参考https://www.cnblogs.com/liangmingshen/p/8906148.html
解决办法:
进入Run -> Edit Configurations -> Parameters,修改参数为2