RuntimeError: cuda runtime error (10) : invalid device ordinal at ...
出现类似报错一般有两种可能
模型在服务器(多GPU)上训练完成,在自己台式机上(仅一块GPU)测试时报错。这是在load时出了问题。
解决可参考 https://blog.csdn.net/yinhui_zhang/article/details/86572232
https://blog.csdn.net/shincling/article/details/78919282
在进行训练时报错。报错如下:
打开train.sh
下面展示一些 内联代码片。
python -m torch.distrib
原创
2021-01-31 11:23:02 ·
516 阅读 ·
0 评论