pytorch加载模型torch.load报cuda out of memory

当服务器中有多张显卡时可能会出现这个问题。
解决方法参考:
pytorch加载模型torch.load报cuda out of memory解决方案
pytorch在加载模型时CUDA error: out of memory

你可能感兴趣的:(笔记,pytorch,深度学习,人工智能)