跳到主要内容

2 篇博文 含有标签「WSL2」

查看所有标签

WSL2 CUDA禁用共享内存

· 阅读需 4 分钟
RibomBalt
CTF enthusiastist, GeoPhysics PhD, Amateur coder

TL;DR;

  • 对于双显卡设备,当N卡显存不够用时,会使用集显提供的共享显存,会显著降低性能。
  • 可以通过torch.cuda.set_per_process_memory_fraction(1.0, 0)在torch模块中禁用共享显存。
  • 禁用共享显存会导致专用显存报错时直接退出而不是保持运行,出现共享显存问题首先应该调参减小模型显存占用。