当显存不够时未调用共享GPU内存？深度学习训练时当显存不够时未调用共享GPU内存进行训练？_深度学习内存可以做显存

作者：AllinToyou | 2024-02-18 04:30:08

踩

深度学习内存可以做显存

当显存不够时未调用共享GPU内存？深度学习训练时当显存不够时未调用共享GPU内存进行训练？

如在stable-diffusion-webui\modules\shared.py文件中383行位置，也可以通过搜索pin_memory从而寻找到目标点。

将上图的False改为True，即383行位置即可

"pin_memory": OptionInfo(True, "Turn on pin_memory for DataLoader. Makes training slightly faster but can increase memory usage."),

即调用torch.utils.data.DataLoader 方法的 pin_memory 设置成了False或者未设置,改为True即可。

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/AllinToyou/article/detail/104724