2018年3月9日
RuntimeError: cuda runtime error (10) : invalid device ordinal
摘要： This is caused by the unmatching of gpu device number when loading a saved model. torch.load('my_file.pt', map_location=lambda storage, loc: storage) 阅读全文
posted @ 2018-03-09 08:06 蠢材少年 阅读(3663) 评论(0) 推荐(0) 编辑