本篇文章小编给大家分享一下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相关异常处理怎么解决关注<计算机技术网(www.ctvol.com)!!>
本文来自网络收集,不代表计算机技术网立场,如涉及侵权请联系管理员删除。
ctvol管理联系方式QQ:251552304
本文章地址:https://www.ctvol.com/phpttorial/103480.html