Voor het Malinees voetbalelftal speelde Diarra tussen 2001 en 2012 in totaal 64 wedstrijden waarin hij 4 doelpunten maakte. Hij nam deel aan het Afrikaans kampioenschap voetbal 2002, 2004, 2008 en 2010. In 2002 en 2004 werd hij met Mali vierde.