编程 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正则匹配查询港澳通行证办理进度示例分享

Dec 27 Python

Django自定义插件实现网站登录验证码功能

Apr 19 Python

Python编程之string相关操作实例详解

Jul 22 Python

scrapy爬虫完整实例

Jan 25 Python

python爱心表白每天都是浪漫七夕!

Aug 18 Python

python实现公司年会抽奖程序

Jan 22 Python

【python】matplotlib动态显示详解

Apr 11 Python

使用python 将图片复制到系统剪贴中

Dec 13 Python

Python如何操作office实现自动化及win32com.client的运用

Apr 01 Python

Numpy 多维数据数组的实现

Jun 18 Python

Python实时监控网站浏览记录实现过程详解

Jul 14 Python

Python各协议下socket黏包问题原理

Apr 12 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网页游戏学习之Xnova(ogame)源码解读（十）

2014/06/24 PHP

html静态页面调用php文件的方法

2014/11/13 PHP

yii去掉必填项中星号的方法

2015/12/28 PHP

php基于自定义函数记录log日志方法

2017/07/21 PHP

PHP多进程通信-消息队列使用

2019/03/08 PHP

PHP实现随机发扑克牌

2020/04/22 PHP

js字符编码函数区别分析

2011/12/28 Javascript

javascript中onclick(this)用法介绍

2013/04/19 Javascript

Javascript异步编程模型Promise模式详细介绍

2014/05/08 Javascript

js+css绘制颜色动态变化的圈中圈效果

2016/01/27 Javascript

JS 实现倒计时数字时钟效果【附实例代码】

2016/03/30 Javascript

TypeScript入门-接口

2017/03/30 Javascript

Axios学习笔记之使用方法教程

2017/07/21 Javascript

史上最全JavaScript常用的简写技巧（推荐）

2017/08/17 Javascript

vue-router 权限控制的示例代码

2017/09/21 Javascript

vue.js通过路由实现经典的三栏布局实例代码

2018/07/08 Javascript

搭建vue开发环境

2018/07/19 Javascript

JavaScript实现更换背景图片

2019/10/18 Javascript

浅谈django开发者模式中的autoreload是如何实现的

2017/08/18 Python

python递归打印某个目录的内容(实例讲解)

2017/08/30 Python

Python实现二叉树的常见遍历操作总结【7种方法】

2019/03/06 Python

解决python文件双击运行秒退的问题

2019/06/24 Python

Python实现TCP通信的示例代码

2019/09/09 Python

python中查看.db文件中表格的名字及表格中的字段操作

2020/07/07 Python

Python+unittest+requests+excel实现接口自动化测试框架

2020/12/23 Python

HTML5 Canvas画线技巧——实现绘制一个像素宽的细线

2013/08/02 HTML / CSS

html5的自定义data-*属性与jquery的data()方法的使用

2014/07/02 HTML / CSS

四川internet信息高速公路(C#)笔试题

2012/02/29 面试题

Python的两道面试题

2013/06/29 面试题

应届毕业生求职信范例分享

2013/12/17 职场文书

《千年梦圆在今朝》教学反思

2014/02/24 职场文书

旅游局领导班子“四风”问题对照检查材料思想汇报

2014/09/29 职场文书

工厂标语大全

2014/10/06 职场文书

民事纠纷协议书

2016/03/23 职场文书

你有一份《诚信考试承诺书》待领取

2019/11/13 职场文书

Python+Matplotlib+LaTeX玩转数学公式

2022/02/24 Python