Pytorch GPU显存充足却显示out of memory怎么办 如何解决
本篇文章小编给大家分享一下Pytorch GPU显存充足却显示out of memory解决方法,小编觉得挺不错的,现在分享给大家供大家参考,有需要的小伙伴们可以来看看。
在测试一个pytorch代码的时候显示显存不足,但是这个网络框架明明很简单,用CPU跑起来都没有问题,GPU却一直提示out of memory.
想来想去也许是pytorch版本的问题,这个pytorch版本是0.4.1,卸载,然后安装pytorch1.1.0,程序就可以神奇的运行了,不会再有OOM的提示了。具体过程如下:
卸载旧版本pytorch:
conda uninstall pytorch
安装pytorch1.1.0,按照官网上的办法,我的CUDA版本是9.0:
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
大功告成!
PHP之友评论