编程 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使用Flask框架同时上传多个文件的方法

Mar 21 Python

Python RuntimeError: thread.__init__() not called解决方法

Apr 28 Python

基于wxpython实现的windows GUI程序实例

May 30 Python

在阿里云服务器上配置CentOS+Nginx+Python+Flask环境

Jun 18 Python

Swift 3.0在集合类数据结构上的一些新变化总结

Jul 11 Python

Python实现的爬虫功能代码

Jun 24 Python

解决pandas无法在pycharm中使用plot()方法显示图像的问题

May 24 Python

使用50行Python代码从零开始实现一个AI平衡小游戏

Nov 21 Python

在linux下实现 python 监控usb设备信号

Jul 03 Python

Python字符编码转码之GBK,UTF8互转

Feb 09 Python

Python超简单容易上手的画图工具库推荐

May 10 Python

详解Python为什么不用设计模式

Jun 24 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

PHP form 表单传参明细研究

2009/07/17 PHP

PHP关于IE下的iframe跨域导致session丢失问题解决方法

2013/10/10 PHP

phpExcel中文帮助手册之常用功能指南

2014/08/18 PHP

支持png透明图片的php生成缩略图类分享

2015/02/08 PHP

Laravel5.1数据库连接、创建数据库、创建model及创建控制器的方法

2016/03/29 PHP

Yii 2.0实现联表查询加搜索分页的方法示例

2017/08/02 PHP

PHP的mysqli_set_charset()函数讲解

2019/01/23 PHP

js模拟实现Array的sort方法

2007/12/11 Javascript

大家未必知道的Js技巧收藏

2008/04/07 Javascript

jQuery Deferred和Promise创建响应式应用程序详细介绍

2013/03/05 Javascript

JavaScript参数个数可变的函数举例说明

2014/10/10 Javascript

使用Node.js为其他程序编写扩展的基本方法

2015/06/23 Javascript

jquery实现适用于门户站的导航下拉菜单效果代码

2015/08/24 Javascript

jQuery 实现评论等级好评差评特效

2016/05/06 Javascript

使用webpack4编译并压缩ES6代码的方法示例

2019/04/24 Javascript

Vue中引入svg图标的两种方式

2021/01/14 Vue.js

js实现电灯开关效果

2021/01/19 Javascript

Python中使用OpenCV库来进行简单的气象学遥感影像计算

2016/02/19 Python

基于MSELoss()与CrossEntropyLoss()的区别详解

2020/01/02 Python

python 使用while循环输出*组成的菱形实例

2020/04/12 Python

css3和jquery实现的可折叠导航菜单适合放在手机网页的导航菜单

2014/09/02 HTML / CSS

html5教程制作简单画板代码分享

2013/12/04 HTML / CSS

HTML5页面直接调用百度地图API获取当前位置直接导航目的地的实现代码

2018/03/02 HTML / CSS

Html5适配iphoneX刘海屏的简单实现

2019/04/09 HTML / CSS

专科应届生求职信

2013/11/24 职场文书

人事主管岗位职责范本

2013/12/04 职场文书

毕业生个人求职信范文分享

2014/01/05 职场文书

大学毕业生个人自荐信范文

2014/01/08 职场文书

护理专业求职信

2014/06/15 职场文书

认错检讨书

2014/10/02 职场文书

党的群众路线教育实践活动自我剖析材料

2014/10/08 职场文书

成品仓管员岗位职责

2015/04/01 职场文书

2015年班干部工作总结

2015/04/29 职场文书

python代码实现扫码关注公众号登录的实战

2021/11/01 Python

详细介绍MySQL中limit和offset的用法

2022/05/06 MySQL

Python时间操作之pytz模块使用详解

2022/06/14 Python