pytorch cuda reserved memory