Pytorch使用训练好的模型时报错RuntimeError: cuda runtime error (10) : invalid device ordinal
跑了两天的模型,拿出来做预测的时候加载出了问题,报错如下:RuntimeError: CUDA error (10): invalid device ordinal (check_status at /opt/conda/conda-bld/pytorch_1532579805626/work/aten/src/ATen/cuda/detail/CUDAHooks.cpp:36) 最后发现...
原创
2018-10-16 13:53:05 ·
9298 阅读 ·
2 评论