CUDA out of memory

[RamyaRaghuraman]

Hi @adynathos,

CUDA runs out of memory at "eval_obj.run_semseg(dset, b_show=False)"

Is there any way to clear the cache memory in CUDA?

[adynathos]

In 020db5cb63d869db50c2f6fd40c41189143f0f4d I add the ability to:

• change the batch size of the pipeline: joint_pipeline.set_batch_size(2)
• run in half-precision with the apex library: joint_pipeline.init_apex_optimization()

Hopefully these help with the GPU memory.

[Olive-fcj]

Is there any way to limit the gpu id?

[adynathos]

@Olive-fcj Control which GPU is used with CUDA_VISIBLE_DEVICES=n, where n is the id of the GPU starting from 0.

https://developer.nvidia.com/blog/cuda-pro-tip-control-gpu-visibility-cuda_visible_devices/