Pytorch GPU显存充足却显示out of memory的解决方式
作者:imaginist233 时间:2022-05-20 18:56:58
今天在测试一个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
标签:Pytorch,GPU显存,outofmemory
0
投稿
猜你喜欢
教你用压缩技术给SQL Server备份文件瘦身
2009-03-05 14:59:00
python的函数最详解
2022-02-15 17:05:05
Ajax:拥抱JSON,让XML走开
2007-10-12 20:01:00
ASP ,IP地址分段计算
2008-04-13 06:55:00
一文详解如何使用Python批量拼接图片
2023-08-12 04:15:37
Python字节码与程序执行过程详解
2022-01-25 04:45:24
Jsp+Servlet实现文件上传下载 删除上传文件(三)
2023-06-27 16:29:29
使用python绘制常用的图表
2022-02-16 19:47:33
如何通过python实现IOU计算代码实例
2021-07-09 02:06:44
Python调用百度api实现语音识别详解
2023-03-10 05:10:17
详解python安装matplotlib库三种失败情况
2023-08-09 15:03:29
Web开发者的百科全书——Google DocType
2008-07-03 13:06:00
pandas 使用apply同时处理两列数据的方法
2021-09-27 07:35:30
Python字符串中如何去除数字之间的逗号
2023-12-01 22:09:48
Python OpenCV一个窗口中显示多幅图像
2023-12-09 19:38:04
(X)HTML的文档结构
2008-06-30 12:25:00
python中os模块和sys模块的使用详解
2021-08-29 21:42:35
python调用文字识别OCR轻松搞定验证码
2023-02-20 17:10:54
django中只使用ModleForm的表单验证
2021-02-03 15:41:50
python爬虫入门教程之点点美女图片爬虫代码分享
2021-01-14 21:45:07