search

Tramac / awesome-semantic-segmentation-pytorch

Semantic Segmentation on PyTorch (include FCN, PSPNet, Deeplabv3, Deeplabv3+, DANet, DenseASPP, BiSeNet, EncNet, DUNet, ICNet, ENet, OCNet, CCNet, PSANet, CGNet, ESPNet, LEDNet, DFANet)
Apache License 2.0
2.79k stars 580 forks source link

RuntimeError: CUDA out of memory. #188

Open wuwangtingfeng opened 2 years ago

wuwangtingfeng commented 2 years ago

Please,How to solve this problem? RuntimeError: CUDA out of memory. Tried to allocate 74.00 MiB (GPU 0; 1.96 GiB total capacity; 1.46 GiB already allocated; 71.50 MiB free; 38.53 MiB cached) My graphics card is older，is Geforce 920M,Is there any solution，thanks

578223592 commented 7 months ago

显存不够啦，你可以尝试：更换更大现存的gpu、减小batch_size、减小输入的特征图分辨率