一行代码消除 PyTorch 的 CUDA 内存溢出报错，这个 GitHub 项目刚发布就揽星 600+

图片地址：https://img.ithome.com/newsuploadfiles/2021/12/e39c0cbb-e829-423c-a842-6599341f2306.png

多少人用 PyTorch“炼丹”时都会被这个 bug 困扰。CUDA error: out of memory.一般情况下，你得找出当下占显存的没用的程序，然后 kill 掉。如果不行，还需手动调整 batch size 到合适的大小，有点麻烦。现在，有人写了一个 PyTorch wrapper，用一行代码就能“无痛”消除这个 bug。有多厉害？相关项目在 GitHub 才发布没几天就收获了 600 + 星。一行代码解决内存溢出错误软件包名叫 koila，已经上传 PyPI，先安装一下：pip install koila现在，假如你面对这样一个 PyTorch 项目：构建一个神经网络来对 FashionMNIST 数据集中的图像进行分类。先定义 input、label 和 model：# A batch of MNIST