本篇文章小编给大家分享一下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