Pytorch GPU显存充足却显示out of memory的解决方式
作者：imaginist233 发布时间：2022-05-20 18:56:58
标签：Pytorch,GPU显存,outofmemory
今天在测试一个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
大功告成!
来源：https://blog.csdn.net/weixin_43155243/article/details/91855612
0
投稿猜你喜欢
楔子我们用 MySQL 客户端查询数据的时候，是以下面这种格式显示的：内容展示的非常漂亮，而 Python 有一个第三方模块叫 pretty- 数据驱动数据的改变从而驱动自动化测试用例的执行，最终引起测试结果的改变。简单说就是参数化的应用。测试驱动在自动化测试中的应用场景：测试步骤的
- webpack我自己还在摸索学习中，今天给大家分享个用webpack把我们的业务模块分开打包的方法，顺便留个笔记 如何用webpa
- 更换国内源vi /etc/apt/sources.listdeb https://mirrors.aliyun.com/kali kali-
- # 2014.10.15 更新了memo.zip, 网盘的exe：修复：1.隔日启动不能正常加载json，加入：1.隐藏任务栏图标，2.通过
- JSON是一种轻量级的数据交换格式，各种语言都有良好的支持。字典是Python的一种数据结构。可以看成关联数组。有些时候我们需要设计到字典转
- 经常看到有人误删数据，或者误操作，特别是update和
- 例子： rsajax.js： var http_request = false; function makePOSTRequest(url,
- if xxx 和if xxx is None的区别一、 if xxxNone,’’,0,[],{},
- 最近，QQ的办公版本——TIM进行了一次更新升级。本次更新升级大幅修改了界面的样式，看起来更加的清爽、简洁和高效了。这种界面州的先生还是比较
- 前言最近公司为客户重新部署了一套新环境，由我来完成了基础环境的配置，配置过程中总结了一些经验，分享给各位园友使用 curl 命令检查网络拿到
- 1.XpathXpath是一门在XML中查找信息的语言，可用来在XML文档中对元素和属性进行遍历。XQuery和xpoint都是构建于xpa
- 1.删除 1）删除记录 Delete from 表名 where id ='xx' 2）
- 什么是循环？ —> 循环是有着周而复始的运动或变化的规律；在 Python 中，循环的操作也叫做 &lsq
- 提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档前言能用golang就用golang..这配置很麻烦 提示：以下是本篇文章正
- 索引1、 什么是索引索引就是数据表中数据和相应的存储位置的列表，利用索引可以提高在表或视图中的查找数据的速度。2、 索引分类数据库中索引主要
- Keras的.h5模型转成tensorflow的.pb格式模型，方便后期的前端部署。直接上代码from keras.models impor
- 一、numpy是什么？扩展库numpy是Python支持科学计算的重要扩展库，是数据分析和科学计算领域如scipy、pandas、sklea
- 本文实例讲述了python修改操作系统时间的方法。分享给大家供大家参考。具体实现方法如下：#-*- coding:utf-8 -*-impo
- 属性 值 描述 onchange 脚本 当元素改变时执行脚本 onsubmit 脚本 当表单被提交时执行脚本 onreset 脚本 当表单被
