粗暴解决CUDA out of memory的问题
作者:李橡树 时间:2023-05-12 05:40:00
小渣渣复现大佬project发现GPU跑不动,出现如下报错:
RuntimeError: CUDA out of memory.
看下来最简单粗暴方法就是减少batch_size,慢是慢了不止一点点但至少跑得动了!
补充:Pytorch GPU显存充足却显示out of memory解决办法
今天在测试一个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_42507051/article/details/90757508
标签:CUDA,out,of,memory
0
投稿
猜你喜欢
Python的内置数据类型中的数字
2023-12-29 19:36:37
详解Hadoop2.7.2 编译64位源码
2023-08-29 13:41:59
insert select与select into 的用法使用说明
2012-01-05 18:47:58
js表单序列化判断空值的实例
2024-04-22 13:01:15
Python进阶篇之正则表达式常用语法总结
2022-03-27 08:45:18
Go1.18新特性之泛型使用三步曲(小结)
2024-05-03 15:05:52
go自动下载所有的依赖包go module使用详解
2024-04-28 10:49:48
python中使用enumerate函数遍历元素实例
2021-05-08 04:56:41
中文用户名的js检验正则
2022-03-26 03:19:32
golang之数组切片的具体用法
2024-04-29 13:06:43
vue实现登录拦截
2024-04-30 10:25:20
Python动态声明变量赋值代码实例
2023-11-15 03:30:47
Python+OpenCV实现将图像转换为二进制格式
2021-06-25 08:10:33
Python中return函数返回值实例用法
2023-11-19 02:11:36
记一次MySQL的优化案例
2024-01-14 21:32:58
JavaScript Generator异步过度的实现详解
2024-04-29 13:45:26
Mysql双主搭建的方法步骤
2024-01-29 05:57:50
pyqt添加启动等待界面的操作
2023-04-29 02:27:11
Python脚本实时处理log文件的方法
2021-02-23 06:40:32
golang解析yaml文件操作
2024-05-21 10:31:30