RuntimeError: CUDA error: invalid device ordinal (exchangeDevice at /opt/conda/conda-bld/pytorch_154

pytorch出现下面的错误
RuntimeError: CUDA error: invalid device ordinal (exchangeDevice at /opt/conda/conda-bld/pytorch_1549628766161/work/aten/src/ATen/cuda/detail/CUDAGuardImpl.h:28)
很大可能是你GPU参数设置的不正确。比如你只有2个GPU,但是你在训练模型时,把GPU的参数设置成4,就会出现上面的错误。

发布了79 篇原创文章 · 获赞 156 · 访问量 7万+

猜你喜欢

转载自blog.csdn.net/ZHUQIUSHI123/article/details/103056810