PyTorch出现”RuntimeError: CUDA out of memory“的解决方法
在运行以下代码时遇到了“RuntimeError: CUDA out of memory. Tried to allocate 66.00 MiB (GPU 0; 2.00 GiB total capacity; 1.21 GiB already allocated; 19.68 MiB free; 1.23 GiB reserved in total by PyTorch)“的问题。
x = self.conv(x)
解决方法:
参考链接:https://blog.youkuaiyun.com/pursui
原创
2020-11-10 16:25:49 ·
3196 阅读 ·
0 评论