В дальнейшем дивизия участвовала в Белградской и Будапештской операциях. В ходе Будапештской операции дивизия вела работу в полосе 46-й армии, полки дивизии обеспечивали ввод в прорыв 2-го гвардейского механизированного корпуса. С 7 ноября 1944 года дивизия участвует в Апатин-Капошварской, Ондавской и Венской операциях, в штурме Вены, Балатонской и Грацско-Амштеттенской операциях. За образцовое выполнение заданий командования в боях с немецкими захватчиками за освобождение города Вена и проявленные при этом доблесть и мужество Указом Президиума Верховного Совета СССР от 17 мая 1945 года награждена орденом Суворова II степени. Войну дивизия закончила на аэродромах Австрии. Боевой состав дивизии на 1 мая 1945 года насчитывал 86 экипажей Ил-2: