编程 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实现监控windows服务并自动启动服务示例

Apr 17 Python

Python格式化压缩后的JS文件的方法

Mar 05 Python

Python数据分析之获取双色球历史信息的方法示例

Feb 03 Python

利用Python实现在同一网络中的本地文件共享方法

Jun 04 Python

pygame游戏之旅载入小车图片、更新窗口

Nov 20 Python

Python数据结构之栈、队列及二叉树定义与用法浅析

Dec 27 Python

python能做什么 python的含义

Oct 12 Python

Pytorch Tensor的统计属性实例讲解

Dec 30 Python

探秘TensorFlow 和 NumPy 的 Broadcasting 机制

Mar 13 Python

Jupyter Notebook输出矢量图实例

Apr 14 Python

解决Alexnet训练模型在每个epoch中准确率和loss都会一升一降问题

Jun 17 Python

python 模拟登陆github的示例

Dec 04 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

S-2000(1) 外键(1) #{}(1) IC-R9500(1) 克隆(1) 小喇叭(1) 脏读(2) 增删查改(1) 小喇叭开始广播了(1) 温控(1)

You might like

PHP生成HTML静态页面实例代码

2008/08/31 PHP

php 页面执行时间计算代码

2008/12/04 PHP

了解Joomla 这款来自国外的php网站管理系统

2010/03/11 PHP

PHP中使用CURL模拟登录并获取数据实例

2014/07/01 PHP

JavaScript中的对象化编程

2008/01/16 Javascript

jquery实现弹出层效果实例

2015/05/19 Javascript

jQuery简单动画变换效果实例分析

2016/07/04 Javascript

Angularjs的Controller间通信机制实例分析

2016/11/07 Javascript

JavaScript基本语法_动力节点Java学院整理

2017/06/26 Javascript

Vue引入jquery实现平滑滚动到指定位置

2018/05/09 jQuery

Vuex 快速入门(简单易懂)

2018/09/20 Javascript

vue结合element-ui使用示例

2019/01/24 Javascript

C#程序员入门学习微信小程序的笔记

2019/03/05 Javascript

转换layUI的数据表格中的日期格式方法

2019/09/19 Javascript

使用JavaScript获取扫码枪扫描得到的条形码的思路代码详解

2020/06/10 Javascript

JS数组reduce()方法原理及使用技巧解析

2020/07/14 Javascript

vue 中的动态传参和query传参操作

2020/11/09 Javascript

[38:32]完美世界DOTA2联赛循环赛 Forest vs DM 第二场 11.06

2020/11/06 DOTA

使用python实现strcmp函数功能示例

2014/03/25 Python

用python 制作图片转pdf工具

2015/01/30 Python

基于Python实现定时自动给微信好友发送天气预报

2018/10/25 Python

Python设计模式之简单工厂模式实例详解

2019/01/22 Python

详解pyppeteer(python版puppeteer)基本使用

2019/06/12 Python

解决Python正则表达式匹配反斜杠''\''问题

2019/07/17 Python

python批量图片处理简单示例

2019/08/06 Python

使用Pytorch来拟合函数方式

2020/01/14 Python

django执行原始查询sql,并返回Dict字典例子

2020/04/01 Python

用CSS3的box-reflect来制作倒影效果

2016/11/15 HTML / CSS

详解HTML5中download属性的应用

2015/08/06 HTML / CSS

大专毕业生自我评价分享

2013/11/10 职场文书

个人收入证明范本

2014/09/18 职场文书

西安兵马俑导游词

2015/02/02 职场文书

董事长助理岗位职责

2015/02/11 职场文书

庆元旦主持词

2015/07/06 职场文书

值班管理制度范本

2015/08/06 职场文书

Android RecyclerView实现九宫格效果

2022/06/28 Java/Android