rtx 4090 RuntimeError: CUDA out of memory. How many memory is needed

TRI-ML / packnet-sfm

TRI-ML Monocular Depth Estimation Repository

https://tri-ml.github.io/packnet-sfm/

MIT License

1.21k stars 241 forks source link

rtx 4090 RuntimeError: CUDA out of memory. How many memory is needed #250

Closed 942411526 closed 1 year ago

942411526 commented 1 year ago

RuntimeError: CUDA out of memory. Tried to allocate 962.00 MiB (GPU 0; 23.65 GiBry Management and PYTORCH_CUDA_ALLOC_CONF Our batch size is 1 and run CUDA_VISIBLE_DEVICES="2" python scripts/train.py configs/train_omnicam.yaml How many memory is needed