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