Діарра захищав кольори молодіжної та олімпійських збірних Малі. Разом зі збірною Малі U-20 виграв молодіжний чемпіонат світу 2013 року в Туреччині. На молодіжному рівні зіграв у 3 офіційних матчах (проти Парагваю, Греції та Мексики). Викликався також і для участі в молодіжному Кубку африканських націй 2013 року в Алжирі.