Diaby debuteerde op 11 oktober 2014 in het Malinees voetbalelftal, onder bondscoach Henryk Kasperczak. Die nam hem een jaar later ook mee naar het Afrikaans kampioenschap 2015. Hij behoorde ook tot de Malinese ploeg op het Afrikaans kampioenschap 2019.