Pytorch GPU显存充足却显示out of memory的解决方式
更新时间：2020年01月13日 10:14:08 作者：imaginist233
今天小编就为大家分享一篇Pytorch GPU显存充足却显示out of memory的解决方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
今天在测试一个pytorch代码的时候显示显存不足，但是这个网络框架明明很简单，用CPU跑起来都没有问题，GPU却一直提示out of memory.
在网上找了很多方法都行不通，最后我想也许是pytorch版本的问题，原来我的pytorch版本是0.4.1，于是我就把这个版本卸载，然后安装了pytorch1.1.0，程序就可以神奇的运行了，不会再有OOM的提示了。虽然具体原因还不知道为何，这里还是先mark一下，具体过程如下：
卸载旧版本pytorch：
conda uninstall pytorch
安装pytorch1.1.0，按照官网上的办法，我的CUDA版本是9.0：
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
大功告成!
以上这篇Pytorch GPU显存充足却显示out of memory的解决方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了Python下的twisted框架入门指引,twisted框架是一个异步机制的框架,也被许多Python教程所介绍,需要的朋友可以参考下2015-04-04
Pytorch 搭建分类回归神经网络并用GPU进行加速的例子
今天小编就为大家分享一篇Pytorch 搭建分类回归神经网络并用GPU进行加速的例子，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-01-01
今天小编就为大家分享一篇对python 命令的-u参数详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-12-12
终端能到import模块 解决jupyter notebook无法导入的问题
这篇文章主要介绍了在终端能到import模块 而在jupyter notebook无法导入的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2021-03-03
这篇文章主要介绍了python实现企业微信定时发送文本消息的示例代码，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-11-11
本文主要介绍了Django与图表的数据交互的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2022-08-08
今天小编就为大家分享一篇python生成任意频率正弦波方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-02-02
这篇文章主要介绍了通过VS下载的NuGet包如何修改其下载存放路径,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2022-09-09
在本文中我们通过实例给大家讲解了关于Python递归函数的用法以及相关知识点，需要的朋友们学习下。2019-02-02
这篇文章主要介绍了十道Python面试最常问到的问题,都是基础的面试题,包括函数,修饰器,数据结构等,需要的朋友可以参考下2023-04-04
最新评论