编程 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中marshal对象序列化的相关知识

Jul 01 Python

MySQL中表的复制以及大型数据表的备份教程

Nov 25 Python

shelve 用来持久化任意的Python对象实例代码

Oct 12 Python

详解Python中的正则表达式

Jul 08 Python

windows下搭建python scrapy爬虫框架步骤

Dec 23 Python

详解从Django Rest Framework响应中删除空字段

Jan 11 Python

Python之列表实现栈的工作功能

Jan 28 Python

Django处理Ajax发送的Get请求代码详解

Jul 29 Python

pytorch神经网络之卷积层与全连接层参数的设置方法

Aug 18 Python

超全Python图像处理讲解(多模块实现)

Apr 13 Python

Pycharm学生免费专业版安装教程的方法步骤

Sep 24 Python

Python爬虫之爬取哔哩哔哩热门视频排行榜

Apr 28 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+javascript的日历控件

2009/11/19 PHP

CodeIgniter框架URL路由总结

2014/09/03 PHP

php调用自己java程序的方法详解

2016/05/13 PHP

PHP连接MySQL数据库操作代码实例解析

2020/07/11 PHP

基于jQuery的Tab选项框效果代码(插件)

2011/03/01 Javascript

JS焦点图切换，上下翻转

2011/05/12 Javascript

js判断一个元素是否为另一个元素的子元素的代码

2012/03/21 Javascript

JavaScript中检测变量是否存在遇到的一些问题

2013/11/11 Javascript

Html5的placeholder属性(IE兼容)实现代码

2014/08/30 Javascript

jQuery表单域属性过滤器用法分析

2015/02/10 Javascript

JavaScript函数节流概念与用法实例详解

2016/06/20 Javascript

NodeJS远程代码执行

2016/08/28 NodeJs

js控制文本框只能输入中文、英文、数字与指定特殊符号的实现代码

2016/09/09 Javascript

Bootstrap栅格系统使用方法及页面调整变形的解决方法

2017/03/10 Javascript

微信小程序聊天室简单实现

2017/04/19 Javascript

nodejs之get/post请求的几种方式小结

2017/07/26 NodeJs

Nginx设置为Node.js的前端服务器方法总结

2019/03/27 Javascript

[40:12]Liquid vs Chaos 2019国际邀请赛小组赛 BO2 第二场 8.15

2019/08/16 DOTA

Python随手笔记第一篇（2）之初识列表和元组

2016/01/23 Python

Python探索之实现一个简单的HTTP服务器

2017/10/28 Python

pandas string转dataframe的方法

2018/04/11 Python

python队列queue模块详解

2018/04/27 Python

Python中操作各种多媒体,视频、音频到图片的代码详解

2020/06/04 Python

pytorch掉坑记录:model.eval的作用说明

2020/06/23 Python

手机配件第一品牌：ZAGG

2017/05/28 全球购物

西班牙美妆电商：Perfume’s Club（有中文站）

2018/08/08 全球购物

教师党性分析材料

2014/02/04 职场文书

四年级语文教学反思

2014/02/05 职场文书

餐饮收银员岗位职责

2014/02/07 职场文书

庆六一文艺汇演活动方案

2014/08/26 职场文书

法人授权委托书公证范本

2014/09/14 职场文书

实施意见格式范本

2015/06/05 职场文书

CAD实训总结范文

2015/08/03 职场文书

用人单位的规章制度，怎样制定才是有效的？

2019/07/09 职场文书

golang 实现对Map进行键值自定义排序

2021/04/28 Golang

MySQL 用 limit 为什么会影响性能

2021/09/15 MySQL