当服务器中有多张显卡时可能会出现这个问题。
解决方法参考:
pytorch加载模型torch.load报cuda out of memory解决方案
pytorch在加载模型时CUDA error: out of memory
pytorch加载模型torch.load报cuda out of memory
最新推荐文章于 2022-09-11 11:42:51 发布
当服务器中有多张显卡时可能会出现这个问题。
解决方法参考:
pytorch加载模型torch.load报cuda out of memory解决方案
pytorch在加载模型时CUDA error: out of memory