深度学习bug汇总

CUDA out of memory.：

img, msk = img.cuda(non_blocking=True).float(), msk.cuda(non_blocking=True).float() 

问题分析：忘记给显卡清缓存，导致运行几次后空间满了（虽然查内存显示只占用了几个MB）。所以要模型在运行前加一句：

torch.cuda.empty_cache()

mamba_ssm和causal-conv1d详细安装教程：

cd 到对应whl文件夹，之后pip install文件名即可

参考链接：mamba_ssm和causal-conv1d详细安装教程 - 知乎