最新下载
热门教程
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
Pytorch GPU显存充足却显示out of memory怎么办 如何解决
时间：2020-01-13 编辑：袖梨 来源：一聚教程网
本篇文章小编给大家分享一下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
大功告成！
相关文章
- Golang ProtoBuf的基本语法详解 10-20
- Python识别MySQL中的冗余索引解析 10-20
- Python+Pygame绘制小球代码展示 10-18
- Python中的数据精度问题介绍 10-18
- Python随机值生成的常用方法介绍 10-18
- python3解压缩.gz文件分析 09-27