编程 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 相关文章推荐

操作Windows注册表的简单的Python程序制作教程

Apr 07 Python

python统计文本字符串里单词出现频率的方法

May 26 Python

深入解析Python中的__builtins__内建对象

Jun 21 Python

python 实现对文件夹内的文件排序编号

Apr 12 Python

python脚本实现验证码识别

Jun 07 Python

django从请求到响应的过程深入讲解

Aug 01 Python

浅谈Pycharm调用同级目录下的py脚本bug

Dec 03 Python

Python图像处理之颜色的定义与使用分析

Jan 03 Python

Python multiprocess pool模块报错pickling error问题解决方法分析

Mar 20 Python

python根据txt文本批量创建文件夹

Dec 08 Python

Python列表list常用内建函数实例小结

Oct 22 Python

python如何导入依赖包

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

关于拼配咖啡，你要知道

2021/03/03 咖啡文化

PHP集成FCK的函数代码

2008/09/27 PHP

PHP高自定义性安全验证码代码

2011/11/27 PHP

Uncaught exception com_exception with message Failed to create COM object

2012/01/11 PHP

深入PHP异步执行的详解

2013/06/03 PHP

PHP微框架Dispatch简介

2014/06/12 PHP

php递归调用删除数组空值元素的方法

2015/04/28 PHP

TextArea不支持maxlength的解决办法(jquery)

2011/09/13 Javascript

JavaScript中的稀疏数组与密集数组[译]

2012/09/17 Javascript

jquery用offset()方法获得元素的xy坐标

2014/09/06 Javascript

浅谈被jQuery抛弃的函数及替代函数

2015/05/03 Javascript

jQuery UI结合Ajax创建可定制的Web界面

2016/06/22 Javascript

js 弹出对话框(遮罩)透明,可拖动的简单实例

2016/07/11 Javascript

如何在Angular.JS中接收并下载PDF

2016/11/26 Javascript

canvas实现绘制吃豆鱼效果

2017/01/12 Javascript

实例讲解Vue.js中router传参

2018/04/22 Javascript

javascript绘制简单钟表效果

2020/04/07 Javascript

Vuex实现购物车小功能

2020/08/17 Javascript

[02:33]2014DOTA2 TI每日综述 LGD涉险晋级DK闯入胜者组

2014/07/14 DOTA

Python设计模式之观察者模式实例

2014/04/26 Python

python使用urlparse分析网址中域名的方法

2015/04/15 Python

git使用.gitignore设置不生效或不起作用问题的解决方法

2017/06/01 Python

flask-restful使用总结

2018/12/04 Python

Python中shapefile转换geojson的示例

2019/01/03 Python

Pyqt5 基本界面组件之inputDialog的使用

2019/06/25 Python

python生成13位或16位时间戳以及反向解析时间戳的实例

2020/03/03 Python

python3处理word文档实例分析

2020/12/01 Python

Wiggle澳大利亚：自行车、跑步、游泳商店

2020/11/07 全球购物

竟聘演讲稿范文

2013/12/31 职场文书

公司员工检讨书

2014/02/08 职场文书

爱我中华教学反思

2014/04/28 职场文书

节电标语大全

2014/06/23 职场文书

个人职业及收入证明

2014/10/13 职场文书

优秀大学生事迹材料

2014/12/24 职场文书

深度学习小工程练习之垃圾分类详解

2021/04/14 Python

Go调用Rust方法及外部函数接口前置

2022/06/14 Golang