编程 Python

Pytorch GPU显存充足却显示out of memory的解决方式

Posted in Python onJanuary 13, 2020

今天在测试一个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的解决方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持三水点靠木。

Pytorch GPU显存充足却显示out of memory的解决方式

- Author -

imaginist233

声明：登载此文出于传递更多信息之目的，并不意味着赞同其观点或证实其描述。

Python 相关文章推荐

python获取网页状态码示例

Mar 30 Python

pycharm 使用心得（三）Hello world！

Jun 05 Python

用Python脚本来删除指定容量以上的文件的教程

May 04 Python

Python实现爬取需要登录的网站完整示例

Aug 19 Python

python实现大转盘抽奖效果

Jan 22 Python

Python 学习教程之networkx

Apr 15 Python

python,Django实现的淘宝客登录功能示例

Jun 12 Python

解决TensorFlow训练内存不断增长,进程被杀死问题

Feb 05 Python

Python 读取位于包中的数据文件

Aug 07 Python

python实现文件分片上传的接口自动化

Nov 19 Python

pymongo insert_many 批量插入的实例

Dec 05 Python

OpenCV+python实现膨胀和腐蚀的示例

Dec 21 Python

Python开发之基于模板匹配的信用卡数字识别功能

Jan 13 #Python

python中的itertools的使用详解

Jan 13 #Python

python3读取csv文件任意行列代码实例

Jan 13 #Python

pytorch程序异常后删除占用的显存操作

Jan 13 #Python

Python跑循环时内存泄露的解决方法

Jan 13 #Python

PyTorch使用cpu加载模型运算方式

Jan 13 #Python

Python如何读取文件中图片格式

Jan 13 #Python

You might like

站长助手-网站web在线管理程序 v1.0 下载

2007/05/12 PHP

Godaddy空间Zend Optimizer升级方法

2010/05/10 PHP

Linux编译升级php的详细方法

2013/11/04 PHP

PHP对接微信公众平台消息接口开发流程教程

2014/03/25 PHP

php 伪静态之IIS篇

2014/06/02 PHP

PHP网络操作函数汇总

2015/05/18 PHP

php脚本运行时的超时机制详解

2016/02/17 PHP

Linux下源码包安装Swoole及基本使用操作图文详解

2019/04/02 PHP

PHP创建XML的方法示例【基于DOMDocument类及SimpleXMLElement类】

2019/09/10 PHP

在jQuery 1.5中使用deferred对象的代码(翻译)

2011/03/10 Javascript

JS时间选择器兼容IE6,7,8,9

2012/06/26 Javascript

理解javascript中try...catch...finally

2015/12/25 Javascript

为jQuery-easyui的tab组件添加右键菜单功能的简单实例

2016/10/10 Javascript

JS实现仿百度文库评分功能

2017/01/12 Javascript

JS中setTimeout和setInterval的最大延时值详解

2017/02/13 Javascript

vue.js 1.x与2.0中js实时监听input值的变化

2017/03/15 Javascript

利用node.js实现反向代理的方法详解

2017/07/24 Javascript

详解vue-cli与webpack结合如何处理静态资源

2017/09/19 Javascript

bootstrap-table实现表头固定以及列固定的方法示例

2019/03/07 Javascript

Node.js+ELK日志规范的实现

2019/05/23 Javascript

Vue如何实现监听组件原生事件

2020/07/03 Javascript

11个并不被常用但对开发非常有帮助的Python库

2015/03/31 Python

Django的信号机制详解

2017/05/05 Python

Python实现将多个空格换为一个空格.md的方法

2018/12/20 Python

python实现PID算法及测试的例子

2019/08/08 Python

Django+uni-app实现数据通信中的请求跨域的示例代码

2019/10/12 Python

keras得到每层的系数方式

2020/06/15 Python

HTML5通用接口详解

2016/06/12 HTML / CSS

美国廉价机票预订网站：Cheapfaremart

2018/04/28 全球购物

亲子拓展活动方案

2014/02/20 职场文书

机关党员进社区活动总结

2014/07/05 职场文书

2014年秋季开学寄语

2014/08/02 职场文书

匿名检举信范文

2015/03/02 职场文书

《暗黑破坏神2：重制版》本周进行第一轮A测目前可官网进行申请报名

2021/04/07 其他游戏

Python实战之实现简易的学生选课系统

2021/05/25 Python

《地。-关于地球的运动-》单行本第七集上市，小说家朝井辽献上期待又害怕的推荐文

2022/03/31 日漫