Created by: sneaxiy

cudaMemGetInfo would raise OOM error when there is very little GPU memory left. This PR refines the error message of this case.