编程 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 MySQLdb模块连接操作mysql数据库实例

Apr 08 Python

在Python中操作列表之List.append()方法的使用

May 20 Python

Python多线程爬虫简单示例

Mar 04 Python

python简单实现获取当前时间

Aug 27 Python

python if not in 多条件判断代码

Sep 21 Python

Python+OpenCV让电脑帮你玩微信跳一跳

Jan 04 Python

Python分支结构（switch）操作简介

Jan 17 Python

Python设置在shell脚本中自动补全功能的方法

Jun 25 Python

python 实现视频流下载保存MP4的方法

Jan 09 Python

flask的orm框架SQLAlchemy查询实现解析

Dec 12 Python

Python 识别12306图片验证码物品的实现示例

Jan 20 Python

Python迭代器协议及for循环工作机制详解

Jul 14 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

简单的页面缓冲技术

2006/10/09 PHP

PHP生成迅雷、快车、旋风等软件的下载链接代码实例

2014/05/12 PHP

支持中文、字母、数字的PHP验证码

2015/05/04 PHP

一个完整的PHP类包含的七种语法说明

2015/06/04 PHP

AJAX PHP无刷新form表单提交的简单实现(推荐)

2016/09/09 PHP

Laravel中错误与异常处理的用法示例

2018/09/16 PHP

PHP判断当前使用的是什么浏览器(推荐)

2019/10/27 PHP

使用jquery实现select添加实现后台权限添加的效果

2011/05/28 Javascript

js 获取后台的字段改变 checkbox的被选中的状态代码

2013/06/05 Javascript

jquery each的几种常用的使用方法示例

2014/01/21 Javascript

利用jQuery实现可以编辑的表格

2014/05/26 Javascript

对JavaScript客户端应用编程的一些建议

2015/06/24 Javascript

javascript适合移动端的日期时间拾取器

2015/11/10 Javascript

谈谈javascript中使用连等赋值操作带来的问题

2015/11/26 Javascript

JavaScript定时器实现的原理分析

2016/12/06 Javascript

详解百度百科目录导航树小插件

2017/01/08 Javascript

JSONP跨域请求

2017/03/02 Javascript

JavaScript实现移动端轮播效果

2017/06/06 Javascript

vue父子组件的嵌套的示例代码

2017/09/08 Javascript

vue-cli之router基本使用方法详解

2017/10/17 Javascript

node作为中间服务层如何发送请求(发送请求的实现方法详解)

2018/01/02 Javascript

详解angular路由高亮之RouterLinkActive

2018/04/28 Javascript

Python ZipFile模块详解

2013/11/01 Python

python 自定义对象的打印方法

2019/01/12 Python

Python实现把类当做字典来访问

2019/12/16 Python

python使用matplotlib的savefig保存时图片保存不完整的问题

2021/01/08 Python

安装不同版本的tensorflow与models方法实现

2021/02/20 Python

美国领先的宠物用品和宠物食品零售商：Petco

2020/10/28 全球购物

介绍下WebSphere的安全性

2013/01/31 面试题

计算机专业自荐信

2014/05/24 职场文书

一般基层干部群众路线教育实践活动个人对照检查材料

2014/11/04 职场文书

小学班主任评语

2014/12/29 职场文书

旷课检讨书

2015/01/26 职场文书

2015年入党积极分子评语

2015/03/26 职场文书

python OpenCV学习笔记

2021/03/31 Python

MySQL 计算连续登录天数

2022/05/11 MySQL